csalabs commited on
Commit
5168fc9
·
1 Parent(s): 5b26759

Update app.py

Browse files

Changing device to cpu

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def main():
108
  chunk_overlap=300)
109
  text_chunks=text_splitter.split_documents(documents)
110
 
111
- embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',model_kwargs={'device': 'cuda'})
112
 
113
  vector_store=FAISS.from_documents(text_chunks, embeddings)
114
 
 
108
  chunk_overlap=300)
109
  text_chunks=text_splitter.split_documents(documents)
110
 
111
+ embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',model_kwargs={'device': 'cpu'})
112
 
113
  vector_store=FAISS.from_documents(text_chunks, embeddings)
114