OduguSusmitha commited on
Commit
7063427
·
verified ·
1 Parent(s): 18412a3

Updated to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from langchain_community.document_loaders import DirectoryLoader
15
 
16
  # Initialize embeddings and ChromaDB
17
  model_name = "sentence-transformers/all-mpnet-base-v2"
18
- device = "cuda" if torch.cuda.is_available() else "cpu"
19
  model_kwargs = {"device": device}
20
  embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
21
 
 
15
 
16
  # Initialize embeddings and ChromaDB
17
  model_name = "sentence-transformers/all-mpnet-base-v2"
18
+ device = "cpu"
19
  model_kwargs = {"device": device}
20
  embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
21