updates

app.py

@@ -35,7 +35,7 @@ with gr.Blocks(fill_height=True) as demo:
         demo_qwen.render()
     with gr.Tab("Experimental"):
         demo_experimental.render()
-    with gr.Tab("AllenAI/Llama-3.1-Tulu-3-8B"):
+    with gr.Tab("AllenAI"):
         demo_allenai.render()
     with gr.Tab("Meta Llama"):
         demo_sambanova.render()

app_allenai.py

@@ -1,21 +1,28 @@
 import gradio as gr
 import spaces
 import transformers_gradio
-from utils import get_app
 
-# Load Llama model
+# Load models
 llama_demo = gr.load(name="allenai/Llama-3.1-Tulu-3-8B", src=transformers_gradio.registry)
 llama_demo.fn = spaces.GPU()(llama_demo.fn)
 
-# Load OLMo model
 olmo_demo = gr.load(name="akhaliq/olmo-anychat", src="spaces")
 
-# Create combined demo with dropdown
-demo = get_app(
-    models=["allenai/Llama-3.1-Tulu-3-8B", "akhaliq/olmo-anychat"],
-    default_model="allenai/Llama-3.1-Tulu-3-8B",
-    src=lambda name, _: llama_demo if name == "allenai/Llama-3.1-Tulu-3-8B" else olmo_demo
-)
+# Create the interface
+with gr.Blocks() as demo:
+    model_dropdown = gr.Dropdown(
+        choices=["allenai/Llama-3.1-Tulu-3-8B", "akhaliq/olmo-anychat"],
+        value="allenai/Llama-3.1-Tulu-3-8B",
+        label="Select Model"
+    )
+    
+    def chat(message, model_name):
+        if model_name == "allenai/Llama-3.1-Tulu-3-8B":
+            return llama_demo.fn(message)
+        else:
+            return olmo_demo.fn(message)
+
+    chatinterface = gr.ChatInterface(chat, additional_inputs=[model_dropdown])
 
 # Disable API names
 for fn in demo.fns.values():