Spaces:
GIZ
/
Running on CPU Upgrade

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

reader error fix

Browse files
Files changed (1) hide show
  1. utils/semantic_search.py +8 -5
utils/semantic_search.py CHANGED
@@ -250,11 +250,14 @@ def semanticSearchPipeline(documents:List[Document], embedding_model:Text = Non
250
  if 'reader' in st.session_state:
251
  reader = st.session_state['reader']
252
 
253
- else:
254
- if reader_model:
255
- reader = FARMReader(model_name_or_path=reader_model,
256
- top_k = reader_top_k, use_gpu=True)
257
- st.session_state['reader'] = reader
 
 
 
258
 
259
  semanticsearch_pipeline = Pipeline()
260
  semanticsearch_pipeline.add_node(component = querycheck, name = "QueryCheck",
 
250
  if 'reader' in st.session_state:
251
  reader = st.session_state['reader']
252
 
253
+ else:
254
+ if reader_model:
255
+ reader = FARMReader(model_name_or_path=reader_model,
256
+ top_k = reader_top_k, use_gpu=True)
257
+ st.session_state['reader'] = reader
258
+ elif reader_model:
259
+ reader = FARMReader(model_name_or_path=reader_model,
260
+ top_k = reader_top_k, use_gpu=True)
261
 
262
  semanticsearch_pipeline = Pipeline()
263
  semanticsearch_pipeline.add_node(component = querycheck, name = "QueryCheck",