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..."):
|