Hugging Face's logo

Spaces:
schrilax
/
leaf-classifier
Runtime error

App Files Community
schrilax commited on
Commit
2d627c2
•
1 Parent(s): e9572e7

add implementation/interface

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