Update app.py

app.py

@@ -53,7 +53,7 @@ local_embedding = HuggingFaceEmbeddings(
     model_name="Alibaba-NLP/gte-Qwen2-1.5B-instruct",
     model_kwargs={
         "trust_remote_code": True,
-        "device": "cuda"  # or "cuda:0"
+        "device": "gpu"  # or "cuda:0"
     }
 )
 
@@ -179,8 +179,6 @@ multiclass_model = DermNetViT(num_classes=len(multiclass_class_names))
 #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 multilabel_model.load_state_dict(torch.load(multilabel_model_path, map_location="cpu"))
 multiclass_model.load_state_dict(torch.load(multiclass_model_path, map_location="cpu"))
-multilabel_model.to(device)
-multiclass_model.to(device)
 
 
 multilabel_model.eval()