Spaces:

santhoshraghu
/

DermBOT

Sleeping

santhoshraghu commited on Apr 24

Commit

d389cd4

verified ·

1 Parent(s): 613658a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,11 +53,10 @@ local_embedding = HuggingFaceEmbeddings(
     model_name="Alibaba-NLP/gte-Qwen2-1.5B-instruct",
     model_kwargs={
         "trust_remote_code": True,
-        "device": "cpu"
     }
 )
 print(" Qwen2-1.5B local embedding model loaded.")

     model_name="Alibaba-NLP/gte-Qwen2-1.5B-instruct",
     model_kwargs={
         "trust_remote_code": True,
+        "device": "cuda"  # or "cuda:0"
     }
 )
 print(" Qwen2-1.5B local embedding model loaded.")