gauravml-89 commited on
Commit
8618073
·
1 Parent(s): 9d773f2

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+ pipe = pipeline(task="image-classification",
4
+ # model for 22k-category classification
5
+ model="google/vit-base-patch16-224")
6
+ gr.Interface.from_pipeline(pipe,
7
+ title="VIT GOOGLE IMG CLASSIFIER",
8
+ description="Object Recognition ...",
9
+ examples = ['wonder_cat.jpg',
10
+ 'aki_dog.jpg',],
11
+ ).launch(inbrowser=True)