init

Signed-off-by: bitliu <bitliu@tencent.com>

app.py

@@ -175,15 +175,31 @@ EXAMPLES_PII = [
 ]
 
 # ============== Gradio Interface ==============
-with gr.Blocks(title="LLM Semantic Router - Model Playground", theme=gr.themes.Soft()) as demo:
-    gr.Markdown(
-        """
-        # 🚀 LLM Semantic Router - Model Playground
+custom_css = """
+.gradio-container {
+    max-width: 1200px !important;
+}
+.header-logo {
+    display: flex;
+    align-items: center;
+    gap: 20px;
+    margin-bottom: 10px;
+}
+.header-logo img {
+    height: 50px;
+}
+"""
 
-        Test our suite of ModernBERT-based classifiers for LLM safety and routing.
-        Select a tab below to try each model.
-        """
-    )
+with gr.Blocks(title="LLM Semantic Router - Model Playground", css=custom_css) as demo:
+    with gr.Row():
+        gr.Image("vllm-logo.png", show_label=False, height=60, width=120,
+                 container=False, show_download_button=False)
+        gr.Markdown(
+            """
+            # 🚀 LLM Semantic Router
+            **Model Playground** - ModernBERT Classifiers for LLM Safety & Routing
+            """
+        )
 
     with gr.Tabs():
         # Tab 1: Fact Check
@@ -256,12 +272,17 @@ with gr.Blocks(title="LLM Semantic Router - Model Playground", theme=gr.themes.S
             ner_btn.click(classify_tokens, ner_input, [ner_output, ner_highlight])
             ner_input.submit(classify_tokens, ner_input, [ner_output, ner_highlight])
 
-    gr.Markdown(
+    gr.HTML(
         """
-        ---
-        **Models**: [LLM-Semantic-Router](https://huggingface.co/LLM-Semantic-Router) |
-        **Architecture**: ModernBERT |
-        **GitHub**: [vllm-project/semantic-router](https://github.com/vllm-project/semantic-router)
+        <hr style="margin-top: 30px;">
+        <div style="text-align: center; color: #666; font-size: 14px;">
+            <p>
+                <strong>Models</strong>: <a href="https://huggingface.co/LLM-Semantic-Router" target="_blank">LLM-Semantic-Router</a> |
+                <strong>Architecture</strong>: ModernBERT |
+                <strong>GitHub</strong>: <a href="https://github.com/vllm-project/semantic-router" target="_blank">vllm-project/semantic-router</a>
+            </p>
+            <p>Part of the <a href="https://github.com/vllm-project/vllm" target="_blank">vLLM</a> ecosystem</p>
+        </div>
         """
     )
 

requirements.txt

@@ -1,4 +1,4 @@
 torch
 transformers>=4.56.0
-gradio>=4.0.0
+gradio>=5.0.0