fixed label error
Browse files
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.
|
15 |
examples = [
|
16 |
"examples/healthy.jpg",
|
17 |
"examples/healthy2.jpg",
|
18 |
"examples/peacock_spot.jpeg",
|
19 |
]
|
20 |
demo = gr.Interface(
|
21 |
-
classify_health,
|
|
|
|
|
|
|
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)
|