Update app.py

app.py

@@ -37,5 +37,5 @@ def predict(img):
     return prediction
     
     
-iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(), outputs=gr.outputs.Label())
+iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(), outputs=gr.outputs.Label(prediction))
 iface.launch()