capofwesh20 commited on
Commit
a852b94
·
1 Parent(s): baf0fd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import datasets
2
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
3
-
4
 
5
  feature_extractor = AutoFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
6
  dataset = datasets.load_dataset("beans")
 
1
  import datasets
2
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
3
+ import gradio as gr
4
 
5
  feature_extractor = AutoFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
6
  dataset = datasets.load_dataset("beans")