Update app.py

app.py

@@ -34,69 +34,66 @@ def generate_image(prompt, height, width, steps, scale, seed):
     except Exception as e:
         return f"An error occurred: {e}"
 
-# Define the input components
-prompt_input = gr.inputs.Textbox(
-    lines=2, 
-    placeholder="Enter your prompt here...", 
-    label="Prompt"
-)
-
-height_input = gr.inputs.Slider(
-    minimum=256, 
-    maximum=2048, 
-    step=64, 
-    default=1024, 
-    label="Image Height"
-)
-
-width_input = gr.inputs.Slider(
-    minimum=256, 
-    maximum=2048, 
-    step=64, 
-    default=1024, 
-    label="Image Width"
-)
-
-steps_input = gr.inputs.Slider(
-    minimum=1, 
-    maximum=50, 
-    step=1, 
-    default=8, 
-    label="Inference Steps"
-)
-
-scale_input = gr.inputs.Slider(
-    minimum=1.0, 
-    maximum=10.0, 
-    step=0.1, 
-    default=3.5, 
-    label="Guidance Scale"
-)
-
-seed_input = gr.inputs.Number(
-    default=3413, 
-    label="Seed", 
-    precision=0
-)
-
-# Define the output component
-image_output = gr.outputs.Image(label="Generated Image")
-
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[prompt_input, height_input, width_input, steps_input, scale_input, seed_input],
-    outputs=image_output,
-    title="Hyper-FLUX-8Steps-LoRA Image Generator",
-    description="Generate images from text prompts using the Hyper-FLUX-8Steps-LoRA model.",
-    examples=[
-        ["A serene landscape with mountains and a river", 1024, 1024, 8, 3.5, 42],
-        ["A futuristic city skyline at sunset", 1024, 1024, 8, 3.5, 1234],
-        ["An abstract painting with vibrant colors", 1024, 1024, 8, 3.5, 5678],
-    ],
-    allow_flagging="never"
-)
+# Define the Gradio interface using the updated API
+with gr.Blocks() as demo:
+    gr.Markdown("# Hyper-FLUX-8Steps-LoRA Image Generator")
+    gr.Markdown(
+        """
+        Enter a text prompt and adjust the parameters to generate an image using the Hyper-FLUX-8Steps-LoRA model.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(
+                label="Prompt",
+                placeholder="Enter your image description here...",
+                lines=2
+            )
+            height = gr.Slider(
+                label="Height",
+                minimum=256,
+                maximum=2048,
+                step=64,
+                value=1024
+            )
+            width = gr.Slider(
+                label="Width",
+                minimum=256,
+                maximum=2048,
+                step=64,
+                value=1024
+            )
+            steps = gr.Slider(
+                label="Steps",
+                minimum=1,
+                maximum=50,
+                step=1,
+                value=8
+            )
+            scale = gr.Slider(
+                label="Scale",
+                minimum=1.0,
+                maximum=10.0,
+                step=0.1,
+                value=3.5
+            )
+            seed = gr.Number(
+                label="Seed",
+                value=3413,
+                precision=0
+            )
+            generate_btn = gr.Button("Generate Image")
+        
+        with gr.Column():
+            output = gr.Image(label="Generated Image")
+    
+    generate_btn.click(
+        fn=generate_image,
+        inputs=[prompt, height, width, steps, scale, seed],
+        outputs=output
+    )
 
-# Launch the interface
+# Launch the Gradio app
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()