diff --git a/flask_predict_no_batch_t5.py b/flask_predict_no_batch_t5.py index 0d5f039..13885ed 100644 --- a/flask_predict_no_batch_t5.py +++ b/flask_predict_no_batch_t5.py @@ -168,7 +168,8 @@ def one_predict(data_text): @return: ''' if data_text[0] != "": - pre_data = autotitle.generate(data_text[0]) + data_inputs = data_text[0].replace("&", "。") + pre_data = autotitle.generate(data_inputs) else: pre_data = "" data_new = [pre_data] + data_text[1:]