kellyshreeve commited on
Commit
4bb1fe5
·
verified ·
1 Parent(s): 897be32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -13,10 +13,12 @@ def predict(img):
13
  title = 'Natural Landscape Classifier'
14
  description = 'Click an example photo or upload your own image!'
15
  examples = ['farm.jpg', 'lake.jpg', 'solar.jpg', 'neighborhood.jpg']
 
 
16
 
17
  iface = gr.Interface(fn=predict,
18
  inputs=gr.Image(type='pil'),
19
- outputs='label',
20
  title=title,
21
  description=description,
22
  examples=examples)
 
13
  title = 'Natural Landscape Classifier'
14
  description = 'Click an example photo or upload your own image!'
15
  examples = ['farm.jpg', 'lake.jpg', 'solar.jpg', 'neighborhood.jpg']
16
+ outputs = gr.Label(value={labels[i]: 0 for i in range(len(labels))},
17
+ label='Probability of Landscape')
18
 
19
  iface = gr.Interface(fn=predict,
20
  inputs=gr.Image(type='pil'),
21
+ outputs=outputs,
22
  title=title,
23
  description=description,
24
  examples=examples)