Fixed bug #828

PyThaiNLP · Jul 27, 2023 · acf5579 · acf5579
1 parent da86fe2
commit acf5579
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pythainlp/chat/core.py b/pythainlp/chat/core.py
@@ -87,7 +87,7 @@ def chat(self, text:str)->str:
         if self.history!=[]:
             for h,b in self.history:
                 _temp+=self.model.PROMPT_DICT['prompt_chatbot'].format_map({"human":h,"bot":b})+self.model.stop_token
-        _temp+=self.model.PROMPT_DICT['prompt_chatbot'].format_map({"human":human,"bot":""})
+        _temp+=self.model.PROMPT_DICT['prompt_chatbot'].format_map({"human":text,"bot":""})
         _bot = self.model.gen_instruct(_temp)
         self.history.append((text,_bot))
         return _bot