Feat App.py Change Label Immediately

app.py

@@ -47,11 +47,11 @@ def chat(message, chat_history):
     chat_history.append((message, response))
     return "", gr.Chatbot(chat_history, label=MODEL_STRS[cur_index], bubble_full_width=False)
 
-def change_model_index(idx):
+def change_model_index(idx, chatbot):
     global cur_index
     cur_index = idx
-    print(cur_index)
-    return
+    # print(cur_index)
+    return gr.Chatbot(chatbot, label=MODEL_STRS[cur_index], bubble_full_width=False)
 
 if __name__=='__main__':
     theme = prepare_theme()
@@ -69,5 +69,5 @@ if __name__=='__main__':
                     txt = gr.Textbox(show_label=False, placeholder='연애 관련 질문을 입력하세요!', container=False)
 
         txt.submit(chat, [txt, chatbot], [txt, chatbot])
-        rd.select(change_model_index, [rd])
+        rd.select(change_model_index, [rd, chatbot], [chatbot])
     demo.launch(debug=True, share=True)