Update app.py

app.py

@@ -33,7 +33,7 @@ enable_queue=True
 
 gr.Interface(
     fn=predict,
-    inputs=gr.Image(shape=(512, 512)),
+    inputs=gr.Image(),
     outputs=gr.Label(num_top_classes=6),
     title=title,
     description=description,