Spaces:
Runtime error
Runtime error
Commit
·
58884d9
1
Parent(s):
dc5bc0f
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@ import transformers
|
|
| 2 |
import gradio as gr
|
| 3 |
import datasets
|
| 4 |
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
|
|
|
|
| 5 |
|
| 6 |
dataset = datasets.load_dataset('beans')
|
| 7 |
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
import datasets
|
| 4 |
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
|
| 5 |
+
from transformers import ViTFeatureExtractor, ViTForImageClassification
|
| 6 |
|
| 7 |
dataset = datasets.load_dataset('beans')
|
| 8 |
|