DrishtiSharma commited on
Commit
9ae551b
Β·
verified Β·
1 Parent(s): f2ed563

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
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
- return Chroma(persist_directory=VECTOR_DB_PATH, collection_name="deepseek_collection", embedding_function=HuggingFaceEmbeddings(model_name="nomic-ai/modernbert-embed-base"))
 
 
 
 
 
 
 
 
 
 
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..."):