fix error

app.py

@@ -36,13 +36,13 @@ example_images = [
 # Gradio Interface
 interface = gr.Interface(
     fn=predict_image,
-    inputs=gr.Image(type="pil", shape=(224, 224)),
+    inputs=gr.Image(type="pil"),
     outputs=gr.Label(num_top_classes=1, label="Prediction"),
     examples=example_images,
     example_per_page=6,
     title="Animals Classifier",
     description="Upload an image of an animal, and the model will predict it.\n\n**Disclaimer:** This model is trained only on specific animal classes (butterfly, cats, cow, dogs, elephant, horse, monkey, sheep, spider, squirrel) and may not accurately predict animals outside these classes.",
-    allow_flagging="never"
+    image_mode="square"
 )
 
 interface.launch(share=True)