Update app.py

app.py

@@ -15,20 +15,32 @@ examples = [
 demo = gr.load("models/GenAIJake/d3xt3r")
 
 # Customize the interface
-with demo:
+with gr.Blocks() as new_demo:
     gr.Markdown("# D3XT3R Dachshund Image Generator")
-    gr.Examples(examples, inputs=demo.input_components)
     
-    # Add LoRA scale slider
-    lora_scale = gr.Slider(minimum=0.0, maximum=1.0, value=0.8, step=0.05, label="LoRA Influence")
+    with gr.Row():
+        with gr.Column():
+            for component in demo.input_components:
+                component.render()
+            
+            # Add LoRA scale slider
+            lora_scale = gr.Slider(minimum=0.0, maximum=1.0, value=0.8, step=0.05, label="LoRA Influence")
+            
+            generate_btn = gr.Button("Generate")
+        
+        with gr.Column():
+            for component in demo.output_components:
+                component.render()
+    
+    gr.Examples(examples, inputs=demo.input_components)
     
     # Update the generate button to include the lora_scale parameter
-    demo.input_components[-1].click(
+    generate_btn.click(
         fn=generate_image,
         inputs=demo.input_components + [lora_scale],
         outputs=demo.output_components
     )
 
 # Add GPUZero functionality
-demo.queue()
-demo.launch()
+new_demo.queue()
+new_demo.launch()