Update app.py

app.py

@@ -71,12 +71,12 @@ def predict(path):
 
 gr.Interface(
     fn=predict,
-    inputs=gr.Image(),
+    inputs=gr.Image(shape=(224, 224)),
     outputs=gr.Label(num_top_classes=5),
     examples=[
         "Dalbergia oliveri.JPG",
         "Eucalyptus.JPG",
         "Khaya senegalensis.JPG",
         "Syzygium nervosum.JPG",
-           ]
+    ],
 ).launch()