Spaces:
Runtime error
Runtime error
second
Browse files- app.py +2 -2
- requirements.txt +2 -1
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
pipe = pipeline(task="image-classification", model="
|
5 |
|
6 |
-
gr.Interface.from_pipeline(pipe, title="
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
+
pipe = pipeline(task="image-classification", model="imjeffhi/pokemon_classifier")
|
5 |
|
6 |
+
gr.Interface.from_pipeline(pipe, title="Pokemon Classifier", description="A fine-tuned version of ViT-base on a collected set of Pokémon images", allow_flagging="never").launch(inbrowser=True)
|
requirements.txt
CHANGED
@@ -1 +1,2 @@
|
|
1 |
-
transformers
|
|
|
|
1 |
+
transformers
|
2 |
+
torch
|