Update app.py

app.py

@@ -2,6 +2,9 @@ import os
 import gradio as gr
 import numpy as np
 from gradio_client import Client
+from PIL import Image
+import requests
+from io import BytesIO
 
 # Используем переменные окружения для гибкости
 MODEL_ID = os.getenv("MODEL_ID", "KingNish/SDXL-Flash")
@@ -137,7 +140,9 @@ with gr.Blocks(css=css) as demo:
                 use_resolution_binning=True,
                 api_name="/run"
             )
-            results.append((response[0], prompt))  # Возвращаем изображение и подпись
+            image_url = response[0]  # Предполагаем, что в ответе содержится URL изображения
+            image = Image.open(BytesIO(requests.get(image_url).content))  # Загружаем изображение из URL
+            results.append(image)
         return results, seed
 
     gr.on(