Update app.py

app.py

@@ -17,6 +17,14 @@ auth_token = os.getenv("auth_token")
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+def download_image_from_url(url):
+    response = requests.get(url)
+    img = Image.open(BytesIO(response.content))
+    tmp = tempfile.NamedTemporaryFile(delete=False)
+    img.save(tmp, 'PNG')
+    tmp.close()
+    return tmp.name
+
 def fetch_image(url):
     try:
         response = requests.get(url)
@@ -70,7 +78,7 @@ def niji_api(prompt):
         time.sleep(1)
     # Process the final image urls
     image_urls = data['response']['imageUrls']
-    yield [iurl, f"image {idx}/4") for idx, iurl in enumerate(image_urls)]
+    yield [(download_image_from_url(iurl), f"image {idx}/4") for idx, iurl in enumerate(image_urls)]
 
 with gr.Blocks() as demo:
     gr.HTML('''