banning Admitted and Abort

handler.py

@@ -22,9 +22,11 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
 
 
-        # Bad word: id 3070, 10456 corresponds to "(*", and we do not want to output a comment
+        # 3070, 10456, [313, 334] corresponds to "(*", and we do not want to output a comment
         # 13 is a newline character
-        bad_words_ids = [[3070], [313, 334], [10456], [13]]
+        # [1976, 441, 29889] is "Abort."
+        # [2087, 29885, 4430, 29889] is "Admitted."
+        bad_words_ids = [[3070], [313, 334], [10456], [13], [1976, 441, 29889], [2087, 29885, 4430, 29889]]
 
         input_ids = self.tokenizer.encode(inputs, return_tensors="pt")