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; EmbeddingRetriever
+from haystack.nodes import FARMReader, TfidfRetriever, EmbeddingRetriever
 import string
 from markdown import markdown
 from annotated_text import annotation
@@ -38,7 +38,7 @@ def start_haystack(documents_processed):
                 model_format="sentence_transformers")
                 document_store.update_embeddings(retriever)
 
-    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