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
|