Spaces:
Sleeping
Sleeping
Update GPU to CPU
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ def split_pages(pages):
|
|
| 40 |
def create_embedding():
|
| 41 |
embeddings = HuggingFaceEmbeddings(
|
| 42 |
model_name="sentence-transformers/all-MiniLM-L6-v2",
|
| 43 |
-
model_kwargs={'device': '
|
| 44 |
)
|
| 45 |
return embeddings
|
| 46 |
|
|
|
|
| 40 |
def create_embedding():
|
| 41 |
embeddings = HuggingFaceEmbeddings(
|
| 42 |
model_name="sentence-transformers/all-MiniLM-L6-v2",
|
| 43 |
+
model_kwargs={'device': 'cpu'}
|
| 44 |
)
|
| 45 |
return embeddings
|
| 46 |
|