Update app.py

app.py

@@ -1,6 +1,6 @@
+import os
 import gradio as gr
 import numpy as np
-import os
 from gradio_client import Client
 
 # Используем переменные окружения для гибкости
@@ -137,7 +137,7 @@ with gr.Blocks(css=css) as demo:
                 use_resolution_binning=True,
                 api_name="/run"
             )
-            results.append(response[0])  # Assuming response contains image path or URL
+            results.append((response[0], prompt))  # Возвращаем изображение и подпись
         return results, seed
 
     gr.on(