MoritzMMuller commited on
Commit
3ac60b6
·
verified ·
1 Parent(s): 1164227

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,12 +47,12 @@ def load_image_model(token: str):
47
  "image-classification",
48
  feature_extractor=AutoFeatureExtractor.from_pretrained(
49
  MODEL_NAME,
50
- subfolder="Skin_Cancer-Image_Classification",
51
  use_auth_token=token
52
  ),
53
  model=AutoModelForImageClassification.from_pretrained(
54
  MODEL_NAME,
55
- subfolder="Skin_Cancer-Image_Classification",
56
  use_auth_token=token
57
  ),
58
  device=0 # or -1 for CPU
 
47
  "image-classification",
48
  feature_extractor=AutoFeatureExtractor.from_pretrained(
49
  MODEL_NAME,
50
+ #subfolder="Skin_Cancer-Image_Classification",
51
  use_auth_token=token
52
  ),
53
  model=AutoModelForImageClassification.from_pretrained(
54
  MODEL_NAME,
55
+ #subfolder="Skin_Cancer-Image_Classification",
56
  use_auth_token=token
57
  ),
58
  device=0 # or -1 for CPU