mkhojira commited on
Commit
8204bc2
1 Parent(s): f7e14ef
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -69,9 +69,12 @@ def classify_image(img):
69
  return dict(zip(categories, map(float, probs)))
70
 
71
 
72
- image = gr.inputs.Image(shape=(192,192))
73
- label = gr.output.Label()
74
  example = ['examples/acadian_flycatcher.jpeg', 'examples/pacific_loon.jpg', 'examples/yellow_throated_vireo.jpeg']
 
 
75
 
76
- iface = gr.Interface(fn=classify_image, inputs=image, outputs=label)
 
77
  iface.launch(inline=False)
 
69
  return dict(zip(categories, map(float, probs)))
70
 
71
 
72
+ image = gr.Image(shape=(192,192))
73
+ label = gr.Label()
74
  example = ['examples/acadian_flycatcher.jpeg', 'examples/pacific_loon.jpg', 'examples/yellow_throated_vireo.jpeg']
75
+ title='North America Bird Classifier'
76
+ description='A north america bird classifier trained on the Caltech-UCSD Birds-200-2011 with fastai.'
77
 
78
+
79
+ iface = gr.Interface(fn=classify_image, inputs=image, outputs=label, title=title,description=description)
80
  iface.launch(inline=False)