wvle commited on
Commit
f3a52cc
1 Parent(s): aad224c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -1,5 +1,14 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
 
 
 
4
  pipe = pipeline("image-classification")
5
- gr.Interface.from_pipeline(pipe).launch()
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ title = 'Interactive Demo for Image Classification'
5
+ des = 'Demo for image classification using image classification pipeline built on ImageNet'
6
+ article = ''
7
+
8
  pipe = pipeline("image-classification")
9
+ gr.Interface.from_pipeline(pipe,
10
+ title=title,
11
+ description=des,
12
+ article=article,
13
+ examples=[['cat.jpg'], ['dog.jpg'], ['elephant.jpg']]
14
+ ).launch()