Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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': '
|
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': 'cuda'})
|
114 |
|
115 |
# Create vector store
|
116 |
vector_store = FAISS.from_documents(text_chunks, embedding=embeddings)
|