Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import datasets
|
2 |
import gradio as gr
|
3 |
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
|
|
|
4 |
|
5 |
dataset = datasets.load_dataset("beans")
|
6 |
|
|
|
1 |
import datasets
|
2 |
import gradio as gr
|
3 |
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
|
4 |
+
import torch
|
5 |
|
6 |
dataset = datasets.load_dataset("beans")
|
7 |
|