Feat Different Placeholder Each Model

app.py

@@ -52,20 +52,19 @@ if __name__=='__main__':
         readme = f.read()
         
     MODELS.append(T5())
+    MODELS[0].placeholder = '연애 관련 질문을 입력하세요!'
     if not LOCAL_TEST:
         MODELS.append(KoAlpaca())
+        MODELS[1].placeholder = '연애 관련 질문을 입력하세요. (KoAlpaca는 추론 시 1분 이상 소요됩니다!)'
 
     with gr.Blocks(theme=prepare_theme()) as demo:
-        with gr.Column():
-            gr.HTML("<h1>KOMUChat</h1>")
-            with gr.Tab("소개"):
-                gr.Markdown(readme)
-            for i in range(len(MODELS)):
-                with gr.Tab(MODEL_STRS[i], id=i):
-                    with gr.Column():
-                        chatbot = gr.Chatbot(label=MODEL_STRS[i], bubble_full_width=False)
-                        txt = gr.Textbox(show_label=False, placeholder='연애 관련 질문을 입력하세요!', container=False, elem_id=i)
-                        
-                        txt.submit(MODELS[i].chat, [txt, chatbot], [txt, chatbot])
+        gr.HTML("<h1>KOMUChat : Korean community-style relationship counseling chabot</h1>")
+        with gr.Tab("소개"):
+            gr.Markdown(readme)
+        for i in range(len(MODELS)):
+            with gr.Tab(MODEL_STRS[i]):
+                chatbot = gr.Chatbot(label=MODEL_STRS[i], bubble_full_width=False)
+                txt = gr.Textbox(show_label=False, placeholder=MODELS[i].placeholder, container=False, elem_id=i)
+                txt.submit(MODELS[i].chat, [txt, chatbot], [txt, chatbot])
 
     demo.launch(debug=True, share=True)

model.py

@@ -1,8 +1,10 @@
 import gradio as gr
 class Model:
     def __init__(self,
-            name:str="Model"):
+            name:str="Model",
+            placeholder:str="Input"):
         self.name = name
+        self.placeholder = placeholder
         self.model = None
         self.tokenizer = None
         self.gen_config = None