Update app.py

app.py

@@ -6,6 +6,7 @@ from pathlib import Path
 from PIL import Image
 import io
 import os
+import requests
 
 import fal_client
 
@@ -135,9 +136,10 @@ def process_image(img, generate_image=True):
             )
             
             # Get the image from the result
-            image_base64 = result["images"][0]["base64"]
-            image_bytes = base64.b64decode(image_base64)
-            generated_image = Image.open(io.BytesIO(image_bytes))
+
+            image_url = result["images"][0]["url"]
+            response = requests.get(image_url)
+            generated_image = Image.open(io.BytesIO(response.content))
 
         else:
             generated_image = None