Fix Gradio 6 ChatInterface: drop type=, use ChatInterface as demo

app.py

@@ -207,62 +207,54 @@ def chat(
         raise gr.Error(f"Generation failed: {errors[0]}")
 
 
-CSS = """
-#col-container { max-width: 920px; margin: 0 auto; }
-.dark .gradio-container { color: var(--body-text-color); }
-"""
-
 DESCRIPTION = f"""
-# Gemma 4 E2B
 Chat demo for Google's edge-sized **Gemma 4** model
-([{BASE_MODEL}]({f"https://huggingface.co/{BASE_MODEL}"})):
+([{BASE_MODEL}](https://huggingface.co/{BASE_MODEL})):
 2.3B effective parameters, 128K context, **text + image + audio**.
 
-This Space runs the instruction-tuned checkpoint
-[`{MODEL_ID}`](https://huggingface.co/{MODEL_ID}) on ZeroGPU.
+Runs the instruction-tuned checkpoint [`{MODEL_ID}`](https://huggingface.co/{MODEL_ID}) on ZeroGPU.
 The [pretrained base](https://huggingface.co/{BASE_MODEL}) is for fine-tuning, not chat.
 
 **Tips:** put the image *before* your question; put audio *after* it.
-Official model card: [google/gemma-4-E2B](https://huggingface.co/google/gemma-4-E2B)
 """
 
 EXAMPLES = [
-    {"text": "Write a short joke about saving RAM."},
-    {
-        "text": "What is shown in this image?",
-        "files": ["examples/golden_gate.png"],
-    },
-    {
-        "text": (
-            "Transcribe the following speech segment in its original language. "
-            "Follow these specific instructions for formatting the answer:\n"
-            "* Only output the transcription, with no newlines.\n"
-            "* When transcribing numbers, write the digits, i.e. write 1.7 and not one point seven, "
-            "and write 3 instead of three."
-        ),
-        "files": ["examples/journal1.wav"],
-    },
+    [{"text": "Write a short joke about saving RAM."}, False, 512],
+    [{"text": "What is shown in this image?", "files": ["examples/golden_gate.png"]}, False, 512],
+    [
+        {
+            "text": (
+                "Transcribe the following speech segment in its original language. "
+                "Follow these specific instructions for formatting the answer:\n"
+                "* Only output the transcription, with no newlines.\n"
+                "* When transcribing numbers, write the digits, i.e. write 1.7 and not one point seven, "
+                "and write 3 instead of three."
+            ),
+            "files": ["examples/journal1.wav"],
+        },
+        False,
+        512,
+    ],
 ]
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, fill_height=True) as demo:
-    with gr.Column(elem_id="col-container"):
-        gr.Markdown(DESCRIPTION)
-        chatbot = gr.ChatInterface(
-            fn=chat,
-            type="messages",
-            multimodal=True,
-            examples=EXAMPLES,
-            additional_inputs=[
-                gr.Checkbox(label="Thinking mode", value=False),
-                gr.Slider(64, 1024, value=512, step=32, label="Max new tokens"),
-            ],
-            additional_inputs_accordion=gr.Accordion("Settings", open=False),
-            textbox=gr.MultimodalTextbox(
-                file_types=["image", ".wav", ".mp3", ".flac", ".ogg", ".m4a", ".mp4", ".webm"],
-                placeholder="Ask Gemma 4 E2B — attach an image or a short audio clip…",
-                show_label=False,
-            ),
-        )
+demo = gr.ChatInterface(
+    fn=chat,
+    multimodal=True,
+    title="Gemma 4 E2B",
+    description=DESCRIPTION,
+    examples=EXAMPLES,
+    cache_examples=False,
+    additional_inputs=[
+        gr.Checkbox(label="Thinking mode", value=False),
+        gr.Slider(64, 1024, value=512, step=32, label="Max new tokens"),
+    ],
+    additional_inputs_accordion="Settings",
+    textbox=gr.MultimodalTextbox(
+        file_types=["image", ".wav", ".mp3", ".flac", ".ogg", ".m4a", ".mp4", ".webm"],
+        placeholder="Ask Gemma 4 E2B — attach an image or a short audio clip…",
+        show_label=False,
+    ),
+)
 
 if __name__ == "__main__":
     demo.launch(mcp_server=True)