diff --git a/fastchat/serve/gradio_web_server.py b/fastchat/serve/gradio_web_server.py
index 8941c6ecb..3963a3c85 100644
--- a/fastchat/serve/gradio_web_server.py
+++ b/fastchat/serve/gradio_web_server.py
@@ -252,21 +252,14 @@ def get_model_list(controller_url, register_api_endpoint_file, vision_arena):
     return visible_models, models
 
 
-def load_demo_single(context: Context, query_params):
-    # default to text models
-    models = context.text_models
-
+def load_demo_single(models, query_params):
     selected_model = models[0] if len(models) > 0 else ""
     if "model" in query_params:
         model = query_params["model"]
         if model in models:
             selected_model = model
 
-    all_models = context.models
-
-    dropdown_update = gr.Dropdown(
-        choices=all_models, value=selected_model, visible=True
-    )
+    dropdown_update = gr.Dropdown(choices=models, value=selected_model, visible=True)
     state = None
     return [state, dropdown_update]