Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ def load_model():
|
|
| 40 |
local_files_only=True,
|
| 41 |
num_labels=15,
|
| 42 |
ignore_mismatched_sizes=True)
|
| 43 |
-
processor = ViTImageProcessor.from_pretrained("
|
| 44 |
|
| 45 |
model.to(device)
|
| 46 |
model.eval()
|
|
|
|
| 40 |
local_files_only=True,
|
| 41 |
num_labels=15,
|
| 42 |
ignore_mismatched_sizes=True)
|
| 43 |
+
processor = ViTImageProcessor.from_pretrained("skin-disease-classification-model-vit/pytorch/default/4")
|
| 44 |
|
| 45 |
model.to(device)
|
| 46 |
model.eval()
|