jye9 commited on
Commit
4b5c69e
·
verified ·
1 Parent(s): 349e4c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -1,10 +1,12 @@
1
- import dradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline(task="image_classification",
5
- model = "micorsoft/beit-base-patch16-384")
6
- gr.Interface.from_pipeline(pipe,
 
 
7
  title="Image Classification",
8
- descrption="Object Reconition app using Microsoft BEIT model",
9
- examples=['egyption_cat.jpg','German_shepherd.jpg',],
10
  ).launch()
 
1
+ import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline(task="image-classification",
5
+ # model that can do category classification
6
+ # https://huggingface.co/microsoft/beit-base-patch16-384
7
+ model = "microsoft/beit-base-patch16-384")
8
+ gr.Interface.from_pipeline(pipe,
9
  title="Image Classification",
10
+ description="Object Recognition using Microsoft BEIT",
11
+ examples = ['egyptian_cat.jpg', 'German_shepherd.jpg',],
12
  ).launch()