Update app.py

app.py

@@ -60,30 +60,36 @@ description = """
 </center></p>
 """
 
-# Gradio interface
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[
-        gr.Textbox(label="Prompt"),
-        gr.Textbox(label="Negative Prompt")
-    ],
-    additional_inputs=[
-        gr.Slider(512, 2048, 1024, step=64, label="Height"),
-        gr.Slider(512, 2048, 1024, step=64, label="Width"),
-        gr.Slider(20, 50, 20, step=1, label="Number of Inference Steps"),
-        gr.Slider(1, 20, 5, step=0.5, label="Guidance Scale"),
-        gr.Slider(1, 4, 1, step=1, label="Number of images per prompt"),
-        gr.Checkbox(label="Use Random Seed", value=True),
-        gr.Number(label="Seed", value=0, precision=0)
-    ],
-    additional_inputs_accordion=gr.Accordion(label="Advanced settings", open=False),
-    outputs=[
-        gr.Gallery(label="Result", elem_id="gallery", show_label=False),
-        gr.Number(label="Seed Used")
-    ],
-    title="Kolors",
-    description=description,
-    theme='bethecloud/storj_theme',
-)
+# Replace your current Gradio interface with this:
+with gr.Blocks(theme='bethecloud/storj_theme', title="Kolors") as iface:
+    gr.Markdown(description)
+    
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(label="Prompt")
+            negative_prompt = gr.Textbox(label="Negative Prompt")
+            
+            with gr.Accordion("Advanced Settings", open=False):
+                height = gr.Slider(512, 2048, 1024, step=64, label="Height")
+                width = gr.Slider(512, 2048, 1024, step=64, label="Width")
+                num_inference_steps = gr.Slider(20, 50, 20, step=1, label="Number of Inference Steps")
+                guidance_scale = gr.Slider(1, 20, 5, step=0.5, label="Guidance Scale")
+                num_images_per_prompt = gr.Slider(1, 4, 1, step=1, label="Number of images per prompt")
+                use_random_seed = gr.Checkbox(label="Use Random Seed", value=True)
+                seed = gr.Number(label="Seed", value=0, precision=0)
+            
+            generate_btn = gr.Button("Generate")
+        
+        with gr.Column():
+            gallery = gr.Gallery(label="Result", show_label=False)
+            seed_used = gr.Number(label="Seed Used")
+    
+    generate_btn.click(
+        fn=generate_image,
+        inputs=[prompt, negative_prompt, height, width, num_inference_steps, 
+                guidance_scale, num_images_per_prompt, use_random_seed, seed],
+        outputs=[gallery, seed_used]
+    )
 
-iface.launch(debug=True)
+# Fix the launch call
+iface.launch(debug=True, share=True)