add embed_model

app.py

@@ -524,11 +524,11 @@ embed_model_id = 'sentence-transformers/all-MiniLM-L6-v2'
 
 # device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
 device = 'cpu'
-# embed_model = HuggingFaceEmbeddings(
-#     model_name=embed_model_id,
-#     model_kwargs={'device': device},
-#     encode_kwargs={'device': device, 'batch_size': 32}
-# )
+embed_model = HuggingFaceEmbeddings(
+    model_name=embed_model_id,
+    model_kwargs={'device': device},
+    encode_kwargs={'device': device, 'batch_size': 32}
+)
 
 embeddings = embeddings_openai