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