diff --git a/viscap/visdialch/data/demo_manager.py b/viscap/visdialch/data/demo_manager.py index 81287ca..3809349 100644 --- a/viscap/visdialch/data/demo_manager.py +++ b/viscap/visdialch/data/demo_manager.py @@ -249,6 +249,10 @@ def respond(self, user_question): answer = TreebankWordDetokenizer().detokenize(answer) + if "n't" in answer: + idx_ = answer.find("n't") + answer = ''.join([answer[i] for i in range(len(answer)) if i != idx_-1]) + # Update the dialog history and return answer self._update(user_question, answer) return answer