Update app.py

app.py

@@ -17,6 +17,9 @@ footer {
 .tabs {
     margin-top: 20px;
 }
+.tab-nav {
+    margin-bottom: 20px;
+}
 .tab-content {
     min-height: 600px;
 }
@@ -26,31 +29,32 @@ footer {
 }
 '''
 
-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)")
+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)") 
 
 demo.launch()