Feat Generalize Special Token Replacement

model.py

@@ -9,6 +9,7 @@ class Model:
         self.tokenizer = None
         self.gen_config = None
         self.INPUT_FORMAT = None
+        self.SPETIAL_TOKENS = ["#화자#", "#청자#", "#(남자)청자#", "#(남자)화자#", "#(여자)청자#", "(여자)화자"]
     
     def generate(self, inputs:str) -> str:
         outputs = inputs
@@ -16,7 +17,7 @@ class Model:
     
     def chat(self, message, chat_history):
         response = self.generate(message)
-        response = response.replace('#화자#', '너')
-        response = response.replace('#청자#', '너')
+        for special_token in self.SPETIAL_TOKENS:
+            response = response.replace(special_token, '너')
         chat_history.append((message, response))
         return "", gr.Chatbot(chat_history, label=self.name, bubble_full_width=False)