Update app.py

app.py

@@ -82,7 +82,7 @@ def load_models():
         print(f"Device set to use {device}")
 
         # Embedding models
-        models['embedding'] = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
+        models['embedding_model'] = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
         models['cross_encoder'] = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2', max_length=512)
         models['semantic_model'] = SentenceTransformer('all-MiniLM-L6-v2')
 
@@ -225,7 +225,7 @@ def translate_text(text, source_to_target='ar_to_en'):
         return text
 
 def embed_query_text(query_text):
-    embedding = models['embedding']
+    embedding = models['embedding_model']
     query_embedding = embedding.encode([query_text])
     return query_embedding