Update app.py

app.py

@@ -211,7 +211,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModel.from_pretrained(model_name)
 
 class BERTSearchEngine:
-    def __init__(self, model, tokenizer):
+    def __init__(self, model, tokenizer, text_database):
         self.raw_procesed_data = [self.preprocess(sample, tokenizer) for sample in text_database]
         self.base = []
         self.retriever = None
@@ -257,7 +257,7 @@ class BERTSearchEngine:
         relevant_indice = np.argmax(cosine_similarities, axis=0)
         return relevant_indice
 
-simple_search_engine = BERTSearchEngine(model, tokenizer)
+simple_search_engine = BERTSearchEngine(model, tokenizer, df["question"])
 simple_search_engine.bert = np.load(bert_base.npy)
 
 # bot bert algorithm without context