Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ app.add_middleware(
|
|
23 |
)
|
24 |
|
25 |
# Charger le processor et le modèle fine-tuné depuis le chemin local
|
26 |
-
local_model_path = r'
|
27 |
processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224")
|
28 |
model = AutoModelForImageClassification.from_pretrained(local_model_path)
|
29 |
# model = AutoModelForImageClassification.from_pretrained("2nzi/vit-finetuned-ucf101")
|
|
|
23 |
)
|
24 |
|
25 |
# Charger le processor et le modèle fine-tuné depuis le chemin local
|
26 |
+
local_model_path = r'./vit-finetuned-ucf101'
|
27 |
processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224")
|
28 |
model = AutoModelForImageClassification.from_pretrained(local_model_path)
|
29 |
# model = AutoModelForImageClassification.from_pretrained("2nzi/vit-finetuned-ucf101")
|