Update app.py

app.py

@@ -49,14 +49,15 @@ collection_name = "ks_collection_1.5BE"
 
 
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 local_embedding = HuggingFaceEmbeddings(
     model_name="Alibaba-NLP/gte-Qwen2-1.5B-instruct",
     model_kwargs={
         "trust_remote_code": True,
-        "device": "gpu"  # or "cuda:0"
+        "device": device
     }
 )
-
 print(" Qwen2-1.5B local embedding model loaded.")