Update app.py

app.py

@@ -17,14 +17,6 @@ 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)
@@ -78,7 +70,7 @@ def niji_api(prompt):
         time.sleep(1)
     # Process the final image urls
     image_urls = data['response']['imageUrls']
-    yield [(download_image_from_url(iurl), f"image {idx}/4") for idx, iurl in enumerate(image_urls)]
+    yield [(iurl, f"image {idx}/4") for idx, iurl in enumerate(image_urls)]
 
 with gr.Blocks() as demo:
     gr.HTML('''