Spaces:
Sleeping
Sleeping
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 |
"""
|