Update app.py

app.py

@@ -27,6 +27,10 @@ def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
 
     return generated_image_bytes
 
+# Функция для обработки действия при нажатии кнопки
+def download_image(bytes):
+    gr.download(bytes, "generated_image.png")
+
 # Создаем интерфейс Gradio
 iface = gr.Interface(
     fn=generate_image,
@@ -36,7 +40,7 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.Image(type="pil", label="Generated Image"),
-        gr.Button("Download Image", onclick=lambda bytes: gr.download(bytes, "generated_image.png"))
+        gr.Button("Download Image", action=download_image)
     ],
     live=True,
     title="Huggingface Image Generator"