Hugging Face's logo

Spaces:
kevincluo
/
leaf_classification
Runtime error

App Files Community
kevincluo commited on
Commit
17f207b
·
1 Parent(s): ead77c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
4
 
5
  dataset = load_dataset("beans")
6
 
 
7
  extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
8
  model = AutoModelForImageClassification.from_pretrained("saved_model_files")
9
 
 
4
 
5
  dataset = load_dataset("beans")
6
 
7
+ image_processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224')
8
  extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
9
  model = AutoModelForImageClassification.from_pretrained("saved_model_files")
10