完成t5单条预测版本,引号中有句号处理

2 years ago · ab4bd477a3
1 changed files with 2 additions and 1 deletions
--- 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:]