Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ persist_dir = os.path.join(LOCAL_BASE, PERSIST_SUBDIR)
|
|
| 50 |
# Recreate the SAME embedding model used to build the index
|
| 51 |
emb = HuggingFaceEmbeddings(
|
| 52 |
model_name="google/embeddinggemma-300m",
|
| 53 |
-
model_kwargs={"device": "
|
| 54 |
encode_kwargs={"normalize_embeddings": True},
|
| 55 |
)
|
| 56 |
|
|
|
|
| 50 |
# Recreate the SAME embedding model used to build the index
|
| 51 |
emb = HuggingFaceEmbeddings(
|
| 52 |
model_name="google/embeddinggemma-300m",
|
| 53 |
+
model_kwargs={"device": "cuda"},
|
| 54 |
encode_kwargs={"normalize_embeddings": True},
|
| 55 |
)
|
| 56 |
|