BugFix: input Context for load_demo instead list

cuauty · cuauty · commit e5da11952a0b · 2024-11-27T11:06:11.000+08:00
New code defines the Context class and use it as input parameter.
diff --git a/fastchat/serve/gradio_web_server.py b/fastchat/serve/gradio_web_server.py
@@ -260,7 +260,8 @@ def load_demo(url_params, request: gr.Request):
             controller_url, args.register_api_endpoint_file, vision_arena=False
         )
 
-    return load_demo_single(models, url_params)
+    current_context = Context(text_models=models, models=models)
+    return load_demo_single(current_context, url_params)
 
 
 def vote_last_response(state, vote_type, model_selector, request: gr.Request):

Original file line number	Diff line number	Diff line change
`@@ -260,7 +260,8 @@ def load_demo(url_params, request: gr.Request):`
`260`	`260`	`controller_url, args.register_api_endpoint_file, vision_arena=False`
`261`	`261`	`)`
`262`	`262`
`263`		`- return load_demo_single(models, url_params)`
	`263`	`+ current_context = Context(text_models=models, models=models)`
	`264`	`+ return load_demo_single(current_context, url_params)`
`264`	`265`
`265`	`266`
`266`	`267`	`def vote_last_response(state, vote_type, model_selector, request: gr.Request):`