fix

utils/summary_utils.py

@@ -7,7 +7,7 @@ from utils.tag_utils import filter_tags
 
 AiSummaryVersion = 2
 MinTagScore = 0.7
-summarization_pipeline = pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")
+summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn")
 en_translation_pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
 tag_gen_pipe_1 = pipeline("text-classification", model="yiyanghkust/finbert-esg-9-categories")
 tag_gen_pipe_2 = pipeline("text-classification", model="dima806/news-category-classifier-distilbert")
@@ -19,7 +19,10 @@ def summarize(id: str, text: str):
         return {
             "ver": AiSummaryVersion
         }
+    print('XXXXXXXXXXXXXXXXXXXXX')
+    print(text)
     summary = get_summarization(text) if len(text) > 2000 else text
+    print(summary)
     translated = get_en_translation(summary)
     tags = get_tags(translated, id)
     tags = filter_tags(tags)
@@ -71,7 +74,6 @@ def get_tags(text: str, id: str):
         tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
         print('XXXXXXXXXXXXXXXXXXXXX')
         print(id)
-        print(text)
         print(tags1, tags2, tags3)
         return tags1 + tags2 + tags3
     except: