Update app.py

app.py

@@ -122,11 +122,9 @@ def process_image(img, generate_image=True):
         if generate_image:
             prompt = random.choice(prompt_templates).format(flower=predicted_class)
             
-            # Initialize FAL Client
-            fal = fal_client(key=os.environ.get("FAL_KEY"))
             
             # Use fal-client to generate image
-            result = fal.subscribe(
+            result = fal_client.subscribe(
                 model_url="fal-ai/flux/dev",
                 prompt=prompt,
                 on_update=on_queue_update