Update app.py

app.py

@@ -5,7 +5,14 @@ from fastai.vision.all import *
 from pathlib import Path
 from PIL import Image
 import io
-from fal import client as fal_client
+import os
+
+# Importing fal-client
+try:
+    from fal.client import Client as FalClient
+except ImportError:
+    print("Error: fal-client module not found. Make sure it's in requirements.txt")
+    # Continue with a fallback if needed
 
 # Dictionary of plant names and their Wikipedia links
 search_terms_wikipedia = {
@@ -120,15 +127,18 @@ def process_image(img, generate_image=True):
         if generate_image:
             prompt = random.choice(prompt_templates).format(flower=predicted_class)
             
+            # Initialize FAL Client
+            fal = FalClient(key=os.environ.get("FAL_KEY"))
+            
             # Use fal-client to generate image
-            result = fal_client.subscribe(
-                "fal-ai/flux/dev",
-                input={"prompt": prompt},
-                on_queue_update=on_queue_update
+            result = fal.subscribe(
+                model_url="fal-ai/flux/dev",
+                prompt=prompt,
+                on_update=on_queue_update
             )
             
             # Get the image from the result
-            image_base64 = result["images"][0]
+            image_base64 = result.images[0]
             image_bytes = base64.b64decode(image_base64)
             generated_image = Image.open(io.BytesIO(image_bytes))
         else: