Update app.py

app.py

@@ -390,6 +390,8 @@ def replay_diffusion(frames, replay_speed):
 
 # Define the Gradio UI
 css = '''.gradio-container > .fillable {max-width: 720px !important}
+h3{margin-top: 1em}
+p{margin-top: 0}
 '''
 with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     gr.Markdown(
@@ -399,7 +401,16 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
         [GitHub](https://github.com/ash80/diffusion-gpt), [Colab](https://colab.research.google.com/github/ash80/diffusion-gpt/blob/master/The_Annotated_Discrete_Diffusion_Models.ipynb)
         """
     )
+
+    generate_button = gr.Button("Generate", variant="primary")
     
+    output_textbox = gr.Textbox(
+        label="Generated Text", 
+        lines=15, 
+        interactive=False, 
+        show_copy_button=True,
+        placeholder="Generation will appear here..."
+    )
     with gr.Row():
         steps_slider = gr.Slider(
             minimum=64, 
@@ -418,16 +429,6 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
             info="Controls the speed of the animation after generation.",
             visible=False
         )
-
-    generate_button = gr.Button("Generate", variant="primary")
-    
-    output_textbox = gr.Textbox(
-        label="Generated Text", 
-        lines=15, 
-        interactive=False, 
-        show_copy_button=True,
-        placeholder="Generation will appear here..."
-    )
     
     diffusion_frames_state = gr.State([])