V1.0完成

2 years ago · 02f379efb4
1 changed files with 1 additions and 1 deletions
--- a/flask_predict_no_batch_t5.py
+++ b/flask_predict_no_batch_t5.py
@ -206,7 +206,7 @@ def predict_data_post_processing(text_list):
    return_list = {}
    sentence_one = []
-    sentence_id = "0"
+    sentence_id = text_list_sentence[0][1]
    for i in text_list_sentence:
        if i[1] == sentence_id:
            sentence_one.append(i[0])