diff --git a/articles_directory_predict_word.py b/articles_directory_predict_word.py index 86d6a0a..d84b871 100644 --- a/articles_directory_predict_word.py +++ b/articles_directory_predict_word.py @@ -61,8 +61,8 @@ nums_word_dict = { "5000": { "prompt": prompt["mulu_title_Level_2_1"], "title_1_nums": "3", - "title_2_nums": ["2","3"], - "title_small_nums": ["3", "2","3"] + "title_2_nums": ["2","4"], + "title_small_nums": ["3", "2","4"] }, "8000": { "prompt": prompt["mulu_title_Level_2_1"], diff --git a/articles_directory_predict_word_online.py b/articles_directory_predict_word_online.py index 87e9bca..6056c55 100644 --- a/articles_directory_predict_word_online.py +++ b/articles_directory_predict_word_online.py @@ -53,10 +53,10 @@ prompt = { nums_word_dict = { "3000": { - "prompt": prompt["mulu_title_Level_2"], + "prompt": prompt["mulu_title_Level_2_1"], "title_1_nums": "3", - "title_2_nums": ["2"], - "title_small_nums": ["3", "2"] + "title_2_nums": ["0", "2"], + "title_small_nums": ["3", "0", "2"] }, "5000": { "prompt": prompt["mulu_title_Level_2_1"], @@ -78,9 +78,9 @@ nums_word_dict = { }, "15000": { "prompt": prompt["mulu_title_Level_3_1"], - "title_1_nums": "5", - "title_2_nums": ["3", "5"], - "title_small_nums": ["5", "3", "5"], + "title_1_nums": "6", + "title_2_nums": ["2", "5"], + "title_small_nums": ["6", "2", "5"], }, "20000": { "prompt": prompt["mulu_title_Level_3_2"],