only return new generated tokens, bad_words = comments

handler.py

@@ -22,7 +22,8 @@ class EndpointHandler():
         """
         inputs = data.pop("inputs", data)
 
-        prediction = self.pipeline(inputs, stopping_criteria=self.stopping_criteria, max_new_tokens=100)
+        # Bad word: id 3070 corresponds to "(*", and we do not want to output a comment
+        prediction = self.pipeline(inputs, stopping_criteria=self.stopping_criteria, max_new_tokens=50, return_full_text=False, bad_words_ids=[[3070]])
         return prediction