chore: fixing gradio components

app.py

@@ -39,9 +39,15 @@ def generate_sketch(prompt, negative_prompt="ugly face, multiple bodies, bad ana
     )
     return result.images[0].convert("RGB")  # Ensure the image is in RGB format
 
+# Gradio Interface
+description = """
+This demo utilizes the SDXL model LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were trained on sketches of Khabib by ritwikraha using DreamBooth.
+"""
 # Setup Gradio interface
 with gr.Blocks() as demo:
-    with gr.Column():
+    gr.HTML("<h1><center>Text-to-Image with SDXL-Lightning ⚡</center></h1>")
+    gr.Markdown(description)
+    with gr.Row():
         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")
         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)