Hackoor commited on
Commit
6cb367d
1 Parent(s): 0ec0c2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ def main():
110
 
111
  # Create embeddings
112
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",
113
- model_kwargs={'device': 'cuda'})
114
 
115
  # Create vector store
116
  vector_store = FAISS.from_documents(text_chunks, embedding=embeddings)
 
110
 
111
  # Create embeddings
112
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2",
113
+ model_kwargs={'device': 'gpu'})
114
 
115
  # Create vector store
116
  vector_store = FAISS.from_documents(text_chunks, embedding=embeddings)