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

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
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()