chore: fixing gradio components

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from diffusers import DiffusionPipeline, AutoencoderKL
 from PIL import Image
-import spaces  # Assuming you're still working within a framework that uses @spaces
+import spaces  
 
 # Initialize the VAE model and Diffusion Pipeline outside the GPU-enabled function for efficiency
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
@@ -48,12 +48,12 @@ with gr.Blocks() as demo:
     gr.HTML("<h1><center>Text-to-Image with SDXL-Lightning ⚡</center></h1>")
     gr.Markdown(description)
     with gr.Group():
-        with gr.Row():
+        with gr.Column():
             prompt_input = gr.Textbox(label="Enter your image prompt", value="a sketch of TOK khabib pointing at another khabib like the spiderman meme, monchrome, pen sketch", scale=8)
             negative_prompt_input = gr.Textbox(label="Enter negative prompt", value="ugly face, multiple bodies, bad anatomy, disfigured, extra fingers", lines=2)
             guidance_scale_slider = gr.Slider(label="Guidance Scale", minimum=1, maximum=5, value=3)
             steps_slider = gr.Slider(label="Number of Inference Steps", minimum=20, maximum=100, value=50)
-            submit_button = gr.Button("Generate Sketch")
+            submit_button = gr.Button("Submit")
         output_image = gr.Image(label="Generated Sketch")
     
     submit_button.click(