Update app.py

app.py

@@ -7,7 +7,8 @@ import shutil
 import numpy as np
 import torch
 from datasets import load_dataset
-from langchain.embeddings import HuggingFaceEmbeddings
+from langchain_community.embeddings import HuggingFaceEmbeddings  # ✅ Correct
+from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_community.vectorstores import Chroma
 from langchain.schema import Document
 from sentence_transformers import SentenceTransformer
@@ -15,10 +16,13 @@ from sklearn.metrics import mean_squared_error, roc_auc_score
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 
+
 # ✅ Load Pretrained Model
 model_name = "bert-base-uncased"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-embedding_model = HuggingFaceEmbeddings(model_name=model_name)
+#embedding_model = HuggingFaceEmbeddings(model_name=model_name)
+embedding_model = HuggingFaceEmbeddings(model_name="models/bert-base-uncased")
+
 embedding_model.client.to(device)
 
 # ✅ Set OpenAI API Key (Replace with your own)