Update app.py

app.py

@@ -28,7 +28,14 @@ def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
     return generated_image_bytes
 
 # Функция для обработки действия при нажатии кнопки
-def download_image(bytes):
+def download_image():
+    bytes = generate_image(
+        prompt=iface.inputs[0].value,
+        negative_prompt=iface.inputs[1].value,
+        sampling_steps=iface.tabs[1].components[0].value,
+        cfg_scale=iface.tabs[1].components[1].value,
+        seed=iface.tabs[1].components[2].value,
+    )
     gr.download(bytes, "generated_image.png")
 
 # Создаем интерфейс Gradio