2nzi commited on
Commit
31216a6
·
verified ·
1 Parent(s): b0dcef4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'.\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")
 
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")