GulAk8902 commited on
Commit
4aad777
·
verified ·
1 Parent(s): b52ab7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("google/vit-base-patch16-224-in21k")
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()