Update app.py

app.py

@@ -176,9 +176,9 @@ multiclass_model_path = hf_hub_download(repo_id="santhoshraghu/DermBOT", filenam
 multilabel_model = SkinViT(num_classes=len(multilabel_class_names))
 multiclass_model = DermNetViT(num_classes=len(multiclass_class_names))
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+#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=device))
+multiclass_model.load_state_dict(torch.load(multiclass_model_path, map_location="cpu"))
 multilabel_model.to(device)
 multiclass_model.to(device)