Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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 |
|