Feat Special Token can be Replaced

README.txt

@@ -1,4 +1,4 @@
-# KOMUCHAT : Korean community-style relationship counseling chabot
+# KOMUCHAT : Korean community-style relationship counseling chatbot
 KOMUChat: 온라인 커뮤니티 스타일의 연애상담챗봇
 ### 개요
 사람들의 실제 대화를 반영하기 위해 한국 커뮤니티 에펨코리아 + 인스티즈에서 수집한 약 3만건의 질문-답변 대화 데이터셋으로 학습시킨 챗봇<br>

app.py

@@ -58,7 +58,7 @@ if __name__=='__main__':
         MODELS[1].placeholder = '연애 관련 질문을 입력하세요. (KoAlpaca는 추론 시 1분 이상 소요됩니다!)'
 
     with gr.Blocks(theme=prepare_theme()) as demo:
-        gr.HTML("<h1>KOMUChat : Korean community-style relationship counseling chabot</h1>")
+        gr.HTML("<h1>KOMUChat : Korean community-style relationship counseling chatbot</h1>")
         with gr.Tab("소개"):
             gr.Markdown(readme)
         for i in range(len(MODELS)):

model.py

@@ -16,5 +16,7 @@ class Model:
     
     def chat(self, message, chat_history):
         response = self.generate(message)
+        response = response.replace('#화자#', '너')
+        response = response.replace('#청자#', '너')
         chat_history.append((message, response))
         return "", gr.Chatbot(chat_history, label=self.name, bubble_full_width=False)