alyxx commited on
Commit
cacfc79
·
1 Parent(s): c908479
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -2,11 +2,10 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  pipe = pipeline(task="image-classification",
5
- # model that can do 22k-category classification
6
- model="microsoft/beit-base-patch16-224-pt22k-ft22k")
7
  gr.Interface.from_pipeline(pipe,
8
- title="22k Image Classification",
9
- description="Object Recognition using Microsoft BEIT",
10
  # examples = ['wonder_cat.jpg', 'aki_dog.jpg',],
11
- article = "Author: <a href=\"https://huggingface.co/rowel\">Rowel Atienza</a>",
12
  ).launch()
 
2
  from transformers import pipeline
3
 
4
  pipe = pipeline(task="image-classification",
5
+ model="kaiku03/vit-base-1stGen-Pokemon-Images")
 
6
  gr.Interface.from_pipeline(pipe,
7
+ title="Whos's that pokemon",
8
+ description="Pokemon Image Classification",
9
  # examples = ['wonder_cat.jpg', 'aki_dog.jpg',],
10
+ # article = "Author: <a href=\"https://huggingface.co/rowel\">Rowel Atienza</a>",
11
  ).launch()