Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ PUBLICATIONS_TO_RETRIEVE = 5
|
|
| 32 |
|
| 33 |
def embedding(
|
| 34 |
model_name: str = "all-MiniLM-L12-v2",
|
| 35 |
-
device: str = "
|
| 36 |
normalize_embeddings: bool = False,
|
| 37 |
) -> langchain_huggingface.HuggingFaceEmbeddings:
|
| 38 |
"""
|
|
|
|
| 32 |
|
| 33 |
def embedding(
|
| 34 |
model_name: str = "all-MiniLM-L12-v2",
|
| 35 |
+
device: str = "cuda",
|
| 36 |
normalize_embeddings: bool = False,
|
| 37 |
) -> langchain_huggingface.HuggingFaceEmbeddings:
|
| 38 |
"""
|