Update appStore/keyword_search.py

appStore/keyword_search.py

@@ -8,24 +8,19 @@ import scripts.clean as clean
 #import needed libraries
 import seaborn as sns
 from pandas import DataFrame
-from sentence_transformers import SentenceTransformer, CrossEncoder, util
-# from keybert import KeyBERT
-from transformers import pipeline
 import matplotlib.pyplot as plt
 import numpy as np
 import streamlit as st
 import pandas as pd 
-from rank_bm25 import BM25Okapi
 from sklearn.feature_extraction import _stop_words
 import string
 from tqdm.autonotebook import tqdm
 import numpy as np
 
-import tempfile
-import sqlite3
 
 #Haystack Components
 @st.cache(hash_funcs={"builtins.SwigPyObject": lambda _: None},allow_output_mutation=True)
+
 def start_haystack(temp.name, file):
     document_store = InMemoryDocumentStore()
     documents = pre.load_document(temp.name, file)
@@ -36,7 +31,6 @@ def start_haystack(temp.name, file):
     pipeline = ExtractiveQAPipeline(reader, retriever)
     return pipeline
 
-
 def ask_question(question):
     prediction = pipeline.run(query=question, params={"Retriever": {"top_k": 10}, "Reader": {"top_k": 5}})
     results = []