Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import fal_client
 import time
+import requests
+from io import BytesIO
+from PIL import Image
 
 def generate_image(api_key, prompt, image_size, num_images, enable_safety_checker, safety_tolerance):
     try:
@@ -13,7 +16,7 @@ def generate_image(api_key, prompt, image_size, num_images, enable_safety_checke
             "Square": "square",
             "Portrait 4:3": "portrait_4_3",
             "Portrait 16:9": "portrait_16_9",
-            "Landscape 4:3": "landscape_4_3",
+            "Landscape 4:3", "landscape_4_3",
             "Landscape 16:9": "landscape_16_9"
         }
         
@@ -50,7 +53,10 @@ def generate_image(api_key, prompt, image_size, num_images, enable_safety_checke
         images = []
         for image_info in result["images"]:
             image_url = image_info["url"]
-            images.append(image_url)
+            # Download the image
+            response = requests.get(image_url)
+            image = Image.open(BytesIO(response.content))
+            images.append(image)
         
         return images if len(images) > 1 else images[0]
     except Exception as e: