santhoshraghu commited on
Commit
a30ff95
·
verified ·
1 Parent(s): f92ed4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -176,9 +176,9 @@ multiclass_model_path = hf_hub_download(repo_id="santhoshraghu/DermBOT", filenam
176
  multilabel_model = SkinViT(num_classes=len(multilabel_class_names))
177
  multiclass_model = DermNetViT(num_classes=len(multiclass_class_names))
178
 
179
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
180
  multilabel_model.load_state_dict(torch.load(multilabel_model_path, map_location="cpu"))
181
- multiclass_model.load_state_dict(torch.load(multiclass_model_path, map_location=device))
182
  multilabel_model.to(device)
183
  multiclass_model.to(device)
184
 
 
176
  multilabel_model = SkinViT(num_classes=len(multilabel_class_names))
177
  multiclass_model = DermNetViT(num_classes=len(multiclass_class_names))
178
 
179
+ #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
180
  multilabel_model.load_state_dict(torch.load(multilabel_model_path, map_location="cpu"))
181
+ multiclass_model.load_state_dict(torch.load(multiclass_model_path, map_location="cpu"))
182
  multilabel_model.to(device)
183
  multiclass_model.to(device)
184