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="facebook/bart-large-cnn")
+summarization_pipeline = pipeline("summarization", model="Falconsai/text_summarization")
 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")
@@ -39,7 +39,7 @@ def get_summarization(text: str):
         print('XXXXXXXXXXXXXXXXXXXXX1')
         print(text)
         # Max / Min number of words
-        result = summarization_pipeline(text, max_length=300, min_length=100, do_sample=False)
+        result = summarization_pipeline(text, max_length=300, min_length=50, do_sample=False)
         print('XXXXXXXXXXXXXXXXXXXXX2')
         print(result)
         return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']