mmek commited on
Commit
1d4cd9f
1 Parent(s): 78027b4

fixed label error

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -11,13 +11,16 @@ def classify_health(input_img):
11
  return dict(zip(categories, map(float, probs)))
12
 
13
 
14
- labels = gr.outputs.Label()
15
  examples = [
16
  "examples/healthy.jpg",
17
  "examples/healthy2.jpg",
18
  "examples/peacock_spot.jpeg",
19
  ]
20
  demo = gr.Interface(
21
- classify_health, gr.Image(height=224, width=224), outputs=labels, examples=examples
 
 
 
22
  )
23
  demo.launch(inline=False)
 
11
  return dict(zip(categories, map(float, probs)))
12
 
13
 
14
+ labels = gr.Label()
15
  examples = [
16
  "examples/healthy.jpg",
17
  "examples/healthy2.jpg",
18
  "examples/peacock_spot.jpeg",
19
  ]
20
  demo = gr.Interface(
21
+ classify_health,
22
+ inputs=gr.Image(height=224, width=224),
23
+ outputs=labels,
24
+ examples=examples,
25
  )
26
  demo.launch(inline=False)