Update appStore/keyword_search.py

appStore/keyword_search.py

@@ -15,7 +15,7 @@ import pandas as pd
 from sklearn.feature_extraction import _stop_words
 from haystack.document_stores import InMemoryDocumentStore
 from haystack.pipelines import ExtractiveQAPipeline
-from haystack.nodes import FARMReader, TfidfRetriever
+from haystack.nodes import FARMReader, TfidfRetriever; EmbeddingRetriever
 import string
 from markdown import markdown
 from annotated_text import annotation
@@ -31,7 +31,7 @@ logger = logging.getLogger(__name__)
 def start_haystack(documents_processed):
     document_store = InMemoryDocumentStore()
     document_store.write_documents(documents_processed)
-    retriever = TfidfRetriever(document_store=document_store)
+    retriever = EmbeddingRetriever(document_store=document_store)
     reader = FARMReader(model_name_or_path="deepset/tinyroberta-squad2", use_gpu=True) #deepset/roberta-base-squad2
     pipeline = ExtractiveQAPipeline(reader, retriever)
     return pipeline