Update app.py

app.py

@@ -32,7 +32,7 @@ def predict_image(image):
     return {"prediction": predict_label, "confidence": float(np.max(predictions))}
 
 # Create the interface
-input_interface = gr.inputs.Image(shape=(256, 256))
+input_interface = gr.inputs.Image(shape=(256, 256, 3))
 output_interface = gr.outputs.Label(num_top_classes=2)
 
 iface = gr.Interface(fn=predict_image, inputs=input_interface, outputs=output_interface)