Spaces:

suresh-subramanian
/

bean-classification

Runtime error

suresh-subramanian commited on Sep 23, 2022

Commit

353a028

1 Parent(s): 58884d9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import transformers
 import gradio as gr
 import datasets
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 from transformers import ViTFeatureExtractor, ViTForImageClassification

 import transformers
 import gradio as gr
 import datasets
+import torch
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 from transformers import ViTFeatureExtractor, ViTForImageClassification