Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -132,7 +132,17 @@ if st.session_state.pdf_loaded and not st.session_state.chunked:
|
|
| 132 |
|
| 133 |
# Step 4: Setup Vectorstore
|
| 134 |
def load_vector_store():
|
| 135 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 136 |
|
| 137 |
if st.session_state.chunked and not st.session_state.vector_created:
|
| 138 |
with st.spinner("Creating vector store..."):
|
|
|
|
| 132 |
|
| 133 |
# Step 4: Setup Vectorstore
|
| 134 |
def load_vector_store():
|
| 135 |
+
try:
|
| 136 |
+
vector_store = Chroma(
|
| 137 |
+
persist_directory=VECTOR_DB_PATH,
|
| 138 |
+
collection_name="deepseek_collection",
|
| 139 |
+
embedding_function=HuggingFaceEmbeddings(model_name="nomic-ai/modernbert-embed-base")
|
| 140 |
+
)
|
| 141 |
+
st.success("✅ Vector store loaded successfully!")
|
| 142 |
+
return vector_store
|
| 143 |
+
except Exception as e:
|
| 144 |
+
st.error(f"❌ Failed to load vector store: {e}")
|
| 145 |
+
return None # Return None if there's an error
|
| 146 |
|
| 147 |
if st.session_state.chunked and not st.session_state.vector_created:
|
| 148 |
with st.spinner("Creating vector store..."):
|