Khaled commited on
Commit
24e4fe0
·
1 Parent(s): 46fb3da

use v4 in the demo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ import seaborn as sns
41
  sns.set_theme(style="whitegrid")
42
 
43
 
44
- model_name = "Khaled/melanoma-v3"
45
 
46
  processor = AutoProcessor.from_pretrained(model_name)
47
  model = AutoModelForImageClassification.from_pretrained(model_name)
 
41
  sns.set_theme(style="whitegrid")
42
 
43
 
44
+ model_name = "Khaled/melanoma-v4"
45
 
46
  processor = AutoProcessor.from_pretrained(model_name)
47
  model = AutoModelForImageClassification.from_pretrained(model_name)