Update app.py

app.py

@@ -53,7 +53,7 @@ def process_image(img):
         "fal-ai/flux/schnell",
         arguments={
             "prompt": f"A detailed, artistic interpretation of {predicted_class} flower in natural setting",
-            "image_size": "256x256"  # Low res for testing
+            "image_size": "square"  # Changed from "256x256" to "square"
         },
         with_logs=True,
         on_queue_update=on_queue_update,