def METHOD_NAME(self, task_type, input_string, input_string2=None, entity1=None, entity2=None, max_predict_len=None):
    max_predict_len = max_predict_len if max_predict_len is not None else self.max_predict_len
    input_text = create_input_with_prompt(task_type, input_string, input_string2, entity1, entity2)
    # tokenize
    encodings = self.tokenizer(input_text, max_seq_len=512)
    encodings = {k: paddle.to_tensor(v) for k, v in encodings.items()}
    outputs = self.model.METHOD_NAME(**encodings, max_length=max_predict_len)[0]
    dec_out = list(map(self.token_decode, outputs))
    output = self.pick_most_common(dec_out)
    print("input_text:", input_text[0])
    print("output:", output)
    print("=" * 50)
    return output