Update app.py

app.py

@@ -2,39 +2,55 @@ import gradio as gr
 import sambanova_gradio
 
 css = '''
-.gradio-container{max-width: 1000px !important}
-h1{text-align:center}
+.gradio-container {
+    max-width: 1200px !important;
+    margin: auto;
+}
+h1 {
+    text-align: center;
+    font-size: 2.5em;
+    margin-bottom: 20px;
+}
 footer {
-    visibility: hidden
+    visibility: hidden;
+}
+.tabs {
+    margin-top: 20px;
+}
+.tab-content {
+    min-height: 600px;
+}
+.chatbot {
+    height: 500px;
+    overflow-y: auto;
 }
 '''
 
-with gr.Blocks(css=css) as demo:
-    with gr.Tab("1B"):
-        gr.load("Meta-Llama-3.2-1B-Instruct", 
-                src=sambanova_gradio.registry, 
-                examples=["Explain quantum gravity to a 5-year old.", "How many R are there in the word Strawberry? Think step by step."])
-        
-    with gr.Tab("3B"):
-        gr.load("Meta-Llama-3.2-3B-Instruct", 
-                src=sambanova_gradio.registry, 
-                examples=["Explain quantum gravity to a 5-year old.", "How many R are there in the word Strawberry? Think step by step."])
-        
-    with gr.Tab("8B"):
-        gr.load("Meta-Llama-3.1-8B-Instruct", 
-                src=sambanova_gradio.registry, 
-                examples=["Explain quantum gravity to a 5-year old.", "How many R are there in the word Strawberry? Think step by step."])
-        
-    with gr.Tab("70B"):
-        gr.load('Meta-Llama-3.1-70B-Instruct-8k', 
-                src=sambanova_gradio.registry, 
-                examples=["Explain quantum gravity to a 5-year old.", "How many R are there in the word Strawberry? Think step by step."])
-        
-    with gr.Tab("405B"):
-        gr.load('Meta-Llama-3.1-405B-Instruct', 
-                src=sambanova_gradio.registry, 
-                examples=["Explain quantum gravity to a 5-year old.", "How many R are there in the word Strawberry? Think step by step."])
-
-    gr.Markdown("### Sambanova API Documentation : [Visit here](https://cloud.sambanova.ai/apis)") 
+with gr.Blocks(css=css, title="Enhanced Language Model Interface") as demo:
+    gr.Markdown("# Language Model Comparison")
+    
+    with gr.Tabs(css_class="tabs") as tabs:
+        for model in ["1B", "3B", "8B", "70B", "405B"]:
+            with gr.Tab(f"{model}", css_class="tab-content"):
+                if model == "405B":
+                    model_name = "Meta-Llama-3.1-405B-Instruct"
+                elif model == "70B":
+                    model_name = "Meta-Llama-3.1-70B-Instruct-8k"
+                elif model == "8B":
+                    model_name = "Meta-Llama-3.1-8B-Instruct"
+                else:
+                    model_name = f"Meta-Llama-3.2-{model}-Instruct"
+                
+                gr.load(
+                    model_name,
+                    src=sambanova_gradio.registry,
+                    examples=[
+                        "Explain quantum gravity to a 5-year old.",
+                        "How many R are there in the word Strawberry? Think step by step."
+                    ],
+                    height=600
+                )
+    
+    gr.Markdown("### Sambanova API Documentation : [Visit here](https://cloud.sambanova.ai/apis)")
 
 demo.launch()