Spaces:
GIZ
/
Running on CPU Upgrade

prashant commited on
Commit
2918b13
1 Parent(s): 2c7f87d

session state error fix

Browse files
Files changed (1) hide show
  1. utils/semantic_search.py +19 -12
utils/semantic_search.py CHANGED
@@ -222,19 +222,26 @@ def semanticSearchPipeline(documents:List[Document], embedding_model:Text = Non
222
  if 'retriever' in st.session_state:
223
  # if st.session_state['retriever']:
224
  retriever = st.session_state['retriever']
225
- else:
226
- if embedding_model:
227
- retriever = loadRetriever(embedding_model = embedding_model,
228
- embedding_model_format=embedding_model_format,
229
- embedding_layer=embedding_layer,
230
- retriever_top_k= retriever_top_k,
231
- document_store = document_store)
232
-
233
- st.session_state['retriever'] = retriever
234
  else:
235
- logging.warning("no streamlit enviornment found, neither embedding model \
236
- provided")
237
- return
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
238
 
239
  document_store.update_embeddings(retriever)
240
  retriever.document_store = document_store
 
222
  if 'retriever' in st.session_state:
223
  # if st.session_state['retriever']:
224
  retriever = st.session_state['retriever']
 
 
 
 
 
 
 
 
 
225
  else:
226
+ if embedding_model:
227
+ retriever = loadRetriever(embedding_model = embedding_model,
228
+ embedding_model_format=embedding_model_format,
229
+ embedding_layer=embedding_layer,
230
+ retriever_top_k= retriever_top_k,
231
+ document_store = document_store)
232
+
233
+ st.session_state['retriever'] = retriever
234
+ else:
235
+ logging.warning("no streamlit enviornment found, neither embedding model \
236
+ provided")
237
+ return
238
+ elif embedding_model:
239
+ retriever = loadRetriever(embedding_model = embedding_model,
240
+ embedding_model_format=embedding_model_format,
241
+ embedding_layer=embedding_layer,
242
+ retriever_top_k= retriever_top_k,
243
+ document_store = document_store)
244
+
245
 
246
  document_store.update_embeddings(retriever)
247
  retriever.document_store = document_store