switched back to langchain

backend2.py

@@ -53,6 +53,7 @@ def load_documents(directory):
     logger.debug("Loaded %d documents in %.2f seconds.", len(documents), end_time - start_time)
     return documents
 
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 def prepare_documents(documents):
     logger.debug("Preparing documents for embedding.")
@@ -67,7 +68,7 @@ def prepare_documents(documents):
         return None
     
     modelPath = "sentence-transformers/all-MiniLM-l6-v2"
-    model_kwargs = {'device':'mps'}
+    model_kwargs = {'device': device}
     encode_kwargs = {'normalize_embeddings': False}
     embeddings = HuggingFaceEmbeddings(model_name=modelPath, model_kwargs=model_kwargs, encode_kwargs=encode_kwargs )