UI fixes

app.py

@@ -158,57 +158,51 @@ def respond(
             yield output
 
 
-with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
-    gr.HTML(
-        """
-        <div class="dz-header">
-            <h1>Dreadzone</h1>
-            <p>CPU-only local chat with NVIDIA Nemotron 3 Nano 4B GGUF.</p>
-        </div>
-        """
-    )
-
-    chatbot = gr.ChatInterface(
-        respond,
+chatbot = gr.ChatInterface(
+    respond,
+    type="messages",
+    title="Dreadzone",
+    description="CPU-only local chat with NVIDIA Nemotron 3 Nano 4B GGUF.",
+    chatbot=gr.Chatbot(
         type="messages",
-        chatbot=gr.Chatbot(
-            type="messages",
-            height=520,
-            show_copy_button=True,
-            container=True,
+        height=420,
+        show_copy_button=True,
+        container=True,
+    ),
+    additional_inputs=[
+        gr.Textbox(
+            value=DEFAULT_SYSTEM_PROMPT,
+            label="System message",
+            lines=3,
         ),
-        additional_inputs=[
-            gr.Textbox(
-                value=DEFAULT_SYSTEM_PROMPT,
-                label="System message",
-                lines=3,
-            ),
-            gr.Slider(
-                minimum=32,
-                maximum=1024,
-                value=384,
-                step=16,
-                label="Max new tokens",
-            ),
-            gr.Slider(
-                minimum=0.0,
-                maximum=1.5,
-                value=0.7,
-                step=0.05,
-                label="Temperature",
-            ),
-            gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),
-            gr.Slider(
-                minimum=1.0,
-                maximum=1.3,
-                value=1.1,
-                step=0.01,
-                label="Repeat penalty",
-            ),
-        ],
-        additional_inputs_accordion=gr.Accordion("Generation settings", open=False),
-    )
+        gr.Slider(
+            minimum=32,
+            maximum=1024,
+            value=384,
+            step=16,
+            label="Max new tokens",
+        ),
+        gr.Slider(
+            minimum=0.0,
+            maximum=1.5,
+            value=0.7,
+            step=0.05,
+            label="Temperature",
+        ),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),
+        gr.Slider(
+            minimum=1.0,
+            maximum=1.3,
+            value=1.1,
+            step=0.01,
+            label="Repeat penalty",
+        ),
+    ],
+    additional_inputs_accordion=gr.Accordion("Generation settings", open=False),
+    css=CSS,
+    theme=gr.themes.Soft(),
+)
 
 
 if __name__ == "__main__":
-    demo.queue(max_size=8).launch()
+    chatbot.queue(max_size=8).launch()