Update app.py

app.py

@@ -202,12 +202,16 @@ 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"
 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):