Update app.py

app.py

@@ -125,9 +125,13 @@ def process_image(img, generate_image=True):
             
             # Use fal-client to generate image
             result = fal_client.subscribe(
-                model_url="fal-ai/flux/dev",
-                prompt=prompt,
-                on_update=on_queue_update
+                "fal-ai/flux/dev",
+                arguments={
+                    "prompt": random.choice(prompt_templates).format(flower=predicted_class),
+                    "image_size": "portrait_4_3"
+                },
+                with_logs=True,
+                on_queue_update=on_queue_update,
             )
             
             # Get the image from the result