init

Signed-off-by: bitliu <[email protected]>

app.py

@@ -175,31 +175,15 @@ EXAMPLES_PII = [
 ]
 
 # ============== Gradio Interface ==============
-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;
-}
-"""
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+        # 🚀 LLM Semantic Router - Model Playground
+        **ModernBERT Classifiers for LLM Safety & Routing**
 
-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
-            """
-        )
+        Part of the [vLLM](https://github.com/vllm-project/vllm) ecosystem
+        """
+    )
 
     with gr.Tabs():
         # Tab 1: Fact Check
@@ -272,17 +256,12 @@ with gr.Blocks(title="LLM Semantic Router - Model Playground", css=custom_css) a
             ner_btn.click(classify_tokens, ner_input, [ner_output, ner_highlight])
             ner_input.submit(classify_tokens, ner_input, [ner_output, ner_highlight])
 
-    gr.HTML(
+    gr.Markdown(
         """
-        <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>
+        ---
+        **Models**: [LLM-Semantic-Router](https://huggingface.co/LLM-Semantic-Router) |
+        **Architecture**: ModernBERT |
+        **GitHub**: [vllm-project/semantic-router](https://github.com/vllm-project/semantic-router)
         """
     )
 

requirements.txt

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