Update app.py

app.py

@@ -21,14 +21,17 @@ dashscope.api_key = YOUR_API_TOKEN
 History = List[Tuple[str, str]]
 Messages = List[Dict[str, str]]
 
+
 def clear_session() -> History:
     return '', []
 
+
 def modify_system_session(system: str) -> str:
     if system is None or len(system) == 0:
         system = default_system
     return system, system, []
 
+
 def history_to_messages(history: History, system: str) -> Messages:
     messages = [{'role': Role.SYSTEM, 'content': system}]
     for h in history:
@@ -46,16 +49,17 @@ def messages_to_history(messages: Messages) -> Tuple[str, History]:
     return system, history
 
 
-def model_chat(query: Optional[str], history: Optional[History], system: str
-) -> Tuple[str, str, History]:
+def model_chat(query: Optional[str], history: Optional[History], system: str, radio: str
+               ) -> Tuple[str, str, History]:
     if query is None:
         query = ''
     if history is None:
         history = []
     messages = history_to_messages(history, system)
     messages.append({'role': Role.USER, 'content': query})
+    label_model = f"qwen2.5-coder-{radio.lower()}-instruct"
     gen = Generation.call(
-        model = "qwen2.5-coder-32b-instruct",
+        model=label_model,
         messages=messages,
         result_format='message',
         stream=True
@@ -67,13 +71,11 @@ def model_chat(query: Optional[str], history: Optional[History], system: str
             system, history = messages_to_history(messages + [{'role': role, 'content': response}])
             yield '', history, system
         else:
-            raise HTTPError( code=404, msg='Request id: %s, Status code: %s, error code: %s, error message: %s' % (
+            raise HTTPError(code=404, msg='Request id: %s, Status code: %s, error code: %s, error message: %s' % (
                 response.request_id, response.status_code,
                 response.code, response.message), hdrs=HTTPMessage(), url='http://example.com', fp=None)
 
 
-
-
 def chiose_radio(radio, system):
     mark_ = gr.Markdown(value=f"<center><font size=8>Qwen2.5-Coder-{radio}-instruct👾</center>")
     chatbot = gr.Chatbot(label=f'Qwen2.5-Coder-{radio.lower()}-instruct')
@@ -98,10 +100,10 @@ def main():
     # 创建两个标签
     with gr.Blocks() as demo:
         with gr.Row():
-            options_coder = ["0.5B", "1.5B", "3B", "7B", "14B", "32B",]
+            options_coder = ["0.5B", "1.5B", "3B", "7B", "14B", "32B", ]
             with gr.Row():
                 radio = gr.Radio(choices=options_coder, label="Qwen2.5-Coder：", value="32B")
-                
+        
         with gr.Row():
             with gr.Accordion():
                 mark_ = gr.Markdown("""<center><font size=8>Qwen2.5-Coder-32B-Instruct Bot👾</center>""")
@@ -119,10 +121,10 @@ def main():
                     sumbit = gr.Button("🚀 Send")
                 
                 textbox.submit(model_chat,
-                            inputs=[textbox, chatbot, system_state],
-                            outputs=[textbox, chatbot, system_input])
+                               inputs=[textbox, chatbot, system_state, radio],
+                               outputs=[textbox, chatbot, system_input])
                 sumbit.click(model_chat,
-                             inputs=[textbox, chatbot, system_state],
+                             inputs=[textbox, chatbot, system_state, radio],
                              outputs=[textbox, chatbot, system_input],
                              concurrency_limit=100)
                 clear_history.click(fn=clear_session,