Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ def classify_image(image):
|
|
41 |
|
42 |
iface = gr.Interface(
|
43 |
fn=classify_image,
|
44 |
-
inputs=
|
45 |
outputs=[
|
46 |
gr.outputs.Textbox(label="Result"),
|
47 |
gr.outputs.Number(label="Confidence"),
|
@@ -55,4 +55,4 @@ iface = gr.Interface(
|
|
55 |
]
|
56 |
)
|
57 |
|
58 |
-
iface.launch()
|
|
|
41 |
|
42 |
iface = gr.Interface(
|
43 |
fn=classify_image,
|
44 |
+
inputs=gr.inputs.Image(shape=(256, 256)),
|
45 |
outputs=[
|
46 |
gr.outputs.Textbox(label="Result"),
|
47 |
gr.outputs.Number(label="Confidence"),
|
|
|
55 |
]
|
56 |
)
|
57 |
|
58 |
+
iface.launch()
|