Spaces:

kevincluo
/

leaf_classification

Runtime error

kevincluo commited on Apr 30, 2023

Commit

37c19ea

1 Parent(s): cfc0bd4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,5 @@
 import datasets
 from datasets import load_dataset, Image
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification, ViTImageProcessor, ViTForImageClassification

 import datasets
+import torch
 from datasets import load_dataset, Image
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification, ViTImageProcessor, ViTForImageClassification