bhardwaj08sarthak commited on
Commit
f198d71
·
verified ·
1 Parent(s): 5e21aed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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": "cpu"}, # use "cuda" if your Space has GPU
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