prashant
commited on
Commit
·
eee9f63
1
Parent(s):
2918b13
reader error fix
Browse files- 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 |
-
|
254 |
-
|
255 |
-
|
256 |
-
|
257 |
-
|
|
|
|
|
|
|
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",
|