Update appStore/keyword_search.py

appStore/keyword_search.py

@@ -32,7 +32,7 @@ def start_haystack(documents_processed):
     document_store = InMemoryDocumentStore()
     document_store.write_documents(documents_processed)
     retriever = TfidfRetriever(document_store=document_store)
-    reader = FARMReader(model_name_or_path="allenai/longformer-large-4096-finetuned-triviaqa", use_gpu=True) #deepset/roberta-base-squad2
+    reader = FARMReader(model_name_or_path="deepset/tinyroberta-squad2", use_gpu=True) #deepset/roberta-base-squad2
     pipeline = ExtractiveQAPipeline(reader, retriever)
     return pipeline
 
@@ -81,6 +81,7 @@ def app():
                                          value="What particular risks does climate change pose for women?",)
         if st.button("Find them."):
         #file = st.file_uploader('Upload PDF File', type=['pdf', 'docx', 'txt'])
+            
             if file is not None:
                 with tempfile.NamedTemporaryFile(mode="wb") as temp:
                     bytes_data = file.getvalue()
@@ -121,5 +122,8 @@ def app():
                         except Exception as e:
                             logging.exception(e)
                     
+            else:
+                st.info("🤔 No document found, please try to upload it at the sidebar!"
+                                    )