Spaces:

kevincluo
/

leaf_classification

Runtime error

kevincluo commited on Apr 30, 2023

Commit

cfc0bd4

1 Parent(s): 17f207b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import datasets
 from datasets import load_dataset, Image
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 dataset = load_dataset("beans")

 import datasets
 from datasets import load_dataset, Image
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification, ViTImageProcessor, ViTForImageClassification
 dataset = load_dataset("beans")