Update app.py

app.py

@@ -22,10 +22,10 @@ def classify_image(inp):
     return confidences
 
 # create Gradio interface
+
 iface = gr.Interface(fn=classify_image, 
-             inputs=gr.Image(),
+             inputs=gr.Image(shape=(224, 224)),
              outputs=gr.Label(num_top_classes=4),
-             examples=["banana.jpg", "battery.jpg", "can.jpg"]
-             )
+             examples=["banana.jpg", 'can.jpg', 'battery.jpg'])
              
 iface.launch(share=True)