Update app.py

app.py

@@ -368,7 +368,7 @@ model_bi_encoder.bert_model.from_pretrained("models/friends_bi_encoder")
 # Load question embeds
 question_embeds = np.load("bi_bert_question.npy")
 
-def chat_bi_bert(question):
+def chat_bi_bert(question, history):
     question = encode(question, model_bi_encoder.bert_tokenizer, model_bi_encoder.bert_model, device).squeeze().cpu().detach().numpy()
     cosine_similarities = cosine_similarity([question], question_embeds).flatten()
     top_indice = np.argmax(cosine_similarities, axis=0)
@@ -398,7 +398,7 @@ class CrossEncoderBert(torch.nn.Module):
 model_cross_encoder = CrossEncoderBert().to(device)
 model_cross_encoder.bert_model.from_pretrained("models/friends_cross_encoder")
 
-def chat_cross_bert(question):
+def chat_cross_bert(question, history):
     
     question_encoded = encode(question, model_bi_encoder.bert_tokenizer, model_bi_encoder.bert_model, device).squeeze().cpu().detach().numpy()
     cosine_similarities = cosine_similarity([question_encoded], question_embeds).flatten()
@@ -442,11 +442,11 @@ def echo(message, history, model):
         return answer
     
     elif model=="Bi-BERT-Encoder":
-        answer = chat_bi_bert(message)
+        answer = chat_bi_bert(message, history)
         return answer
 
     elif model=="Bi+Cross-BERT-Encoder":
-        answer = chat_cross_bert(message)
+        answer = chat_cross_bert(message, history)
         return answer