Update app.py

app.py

@@ -41,7 +41,7 @@ def classify_image(image):
 
 iface = gr.Interface(
     fn=classify_image,
-    inputs="image",
+    inputs=gr.inputs.Image(shape=(256, 256)),
     outputs=[
         gr.outputs.Textbox(label="Result"),
         gr.outputs.Number(label="Confidence"),
@@ -55,4 +55,4 @@ iface = gr.Interface(
     ]
 )
 
-iface.launch()
+iface.launch()