Spaces:
Sleeping
Sleeping
danielhshi8224 commited on
Commit ·
f11ecc4
1
Parent(s): 5c455e2
fix processor
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ print(f"Loading model from: {MODEL_ID}")
|
|
| 40 |
# num_labels=7,
|
| 41 |
# ignore_mismatched_sizes=True
|
| 42 |
# )
|
| 43 |
-
processor=AutoImageProcessor.from_pretrained(
|
| 44 |
model = AutoModelForImageClassification.from_pretrained(MODEL_ID)
|
| 45 |
|
| 46 |
# Load weights
|
|
|
|
| 40 |
# num_labels=7,
|
| 41 |
# ignore_mismatched_sizes=True
|
| 42 |
# )
|
| 43 |
+
processor=AutoImageProcessor.from_pretrained('facebook/convnext-tiny-224')
|
| 44 |
model = AutoModelForImageClassification.from_pretrained(MODEL_ID)
|
| 45 |
|
| 46 |
# Load weights
|