Update app.py

app.py

@@ -28,39 +28,23 @@ def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
     return generated_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,
-    )
+def download_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
+    bytes = generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed)
     gr.download(bytes, "generated_image.png")
 
 # Создаем интерфейс Gradio
 iface = gr.Interface(
-    fn=generate_image,
+    fn=download_image,
     inputs=[
         gr.Textbox(label="Prompt", placeholder="Enter prompt here"),
-        gr.Textbox(label="Negative Prompt", placeholder="Enter negative prompt here")
-    ],
-    outputs=[
-        gr.Image(type="pil", label="Generated Image"),
-        gr.Button("Download Image", onclick=download_image)
-    ],
-    live=True,
-    title="Huggingface Image Generator"
-)
-
-# Добавляем вторую вкладку
-iface.add_tab(
-    "Advanced Settings",
-    [
+        gr.Textbox(label="Negative Prompt", placeholder="Enter negative prompt here"),
         gr.Slider(minimum=1, maximum=30, label="Sampling Steps", default=15),
         gr.Slider(minimum=1, maximum=20, label="CFG Scale", default=10),
         gr.Textbox(label="Seed", placeholder="Enter seed here", default="-1")
-    ]
+    ],
+    outputs=gr.Image(type="pil", label="Generated Image"),
+    live=True,
+    title="Huggingface Image Generator"
 )
 
 # Запускаем интерфейс