Update app.py

app.py

@@ -10,7 +10,8 @@ from langchain import PromptTemplate
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_groq import ChatGroq
 from langchain.vectorstores import Chroma
-
+from langchain.embeddings import HuggingFaceEmbeddings
+import torch
 
 
 
@@ -25,7 +26,9 @@ groq_api_key = os.environ.get("my_groq_api_key")
 # Initialize a ChatGroq object with a temperature of 0 and the "mixtral-8x7b-32768" model.
 llm = ChatGroq(temperature=0, model_name="llama3-70b-8192",api_key=groq_api_key)
 
-
+model_name = "BAAI/bge-m3"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs={'device': device})
 # we run this cell every time
 db = Chroma(embedding_function=embeddings, persist_directory='/Persian Chroma/')