Update app.py

app.py

@@ -202,16 +202,13 @@ def chat_word2vec_context(question, history):
   return answer  
 
 #------------------------------------------------------------------------------------------------#
-from sentence_transformers import SentenceTransformer
-
-model = SentenceTransformer("elastic/multilingual-e5-small-optimized")
  
 # Let's try bert model by elastic and with e5
-model_name = "elastic/multilingual-e5-small-optimized"
+model_name = "distilbert/distilbert-base-uncased"
 device = "cpu"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-# model = AutoModel.from_pretrained(model_name)
+model = AutoModel.from_pretrained(model_name)
 
 class BERTSearchEngine:
     def __init__(self, model, tokenizer):