Feat App.py Change Label

app.py

@@ -4,6 +4,7 @@ from t5 import T5
 from koalpaca import KoAlpaca
 
 LOCAL_TEST = False
+MODEL_STRS = ['T5', 'KoAlpaca']
 MODELS = []
 cur_index = 0
 
@@ -44,7 +45,7 @@ def prepare_theme():
 def chat(message, chat_history):
     response = MODELS[cur_index].generate(message)
     chat_history.append((message, response))
-    return "", chat_history
+    return "", gr.Chatbot(chat_history, label=MODEL_STRS[cur_index], bubble_full_width=False)
 
 def change_model_index(idx):
     global cur_index
@@ -61,11 +62,11 @@ if __name__=='__main__':
 
     with gr.Blocks(theme=theme) as demo:
         with gr.Row():
-            rd = gr.Radio(['T5','KoAlpaca'], value='T5', type='index', label='Model Selection', show_label=True, interactive=True)
+            rd = gr.Radio(MODEL_STRS, value='T5', type='index', label='Model Selection', show_label=True, interactive=True)
             with gr.Column(scale=5): # 챗봇 부분 
                 chatbot = gr.Chatbot(label="T5", bubble_full_width=False)
                 with gr.Row():
-                    txt = gr.Textbox(show_label=False, placeholder='Send a message...', container=False)
+                    txt = gr.Textbox(show_label=False, placeholder='연애 관련 질문을 입력하세요!', container=False)
 
         txt.submit(chat, [txt, chatbot], [txt, chatbot])
         rd.select(change_model_index, [rd])