diff --git a/fastchat/serve/gradio_web_server.py b/fastchat/serve/gradio_web_server.py index 1346790e4..c58f5dd36 100644 --- a/fastchat/serve/gradio_web_server.py +++ b/fastchat/serve/gradio_web_server.py @@ -726,7 +726,7 @@ def build_single_model_ui(models, add_promotion_links=False): ) with gr.Row(): with gr.Accordion( - "🔍 Expand to see 20+ model descriptions", + f"🔍 Expand to see the descriptions of {len(models)} models", open=False, elem_id="model_description_accordion", ): @@ -854,7 +854,7 @@ def build_demo(models): state, model_selector, ], - _js=load_js, + js=load_js, ) return demo diff --git a/pyproject.toml b/pyproject.toml index ded93bde7..9de1931f8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -20,7 +20,7 @@ dependencies = [ [project.optional-dependencies] model_worker = ["accelerate>=0.21", "peft", "sentencepiece", "torch", "transformers>=4.31.0", "protobuf"] -webui = ["gradio"] +webui = ["gradio>=4.10"] train = ["einops", "flash-attn>=2.0", "wandb"] llm_judge = ["openai<1", "anthropic>=0.3", "ray"] dev = ["black==23.3.0", "pylint==2.8.2"]