fix null article exception

app.py

@@ -17,12 +17,19 @@ def get_summary(input_text):
     summary_model = BartForConditionalGeneration.from_pretrained("ainize/kobart-news")
     input_ids = tokenizer.encode(input_text, return_tensors="pt")
     summary_text_ids = summary_model.generate(
-        input_ids=input_ids,
-        length_penalty=0.0,
-        max_length=1000,
-        min_length=56,
-        num_beams=2,
+        length_penalty=0,
+        top_p=0.9,
+        max_length=128,
+        min_length=12,
+        num_beams=4,
     )
+    #   "task_specific_params": {
+    #     "summarization": {
+    #       "length_penalty": 1.0,
+    #       "max_length": 128,
+    #       "min_length": 12,
+    #       "num_beams": 4
+    #    }
     return tokenizer.decode(summary_text_ids[0], skip_special_tokens=True)
 
 
@@ -60,7 +67,8 @@ class news_collector:
 
         for news_url in news_links:
             article = self.get_new_parser(news_url)
-            self.examples_text.append([get_summary(article.text[:3000]), news_url])
+            if article.text:
+                self.examples_text.append([get_summary(article.text[:1500]), news_url])
 
 
 
@@ -97,8 +105,8 @@ with gr.Blocks(theme='pseudolab/huggingface-korea-theme') as demo:
 
     with gr.Tab("데모"):
         Link_TXT = gr.Textbox(label="뉴스 내용", placeholder = "뉴스 기사 내용을 입력하세요.")
-        # gr.load("models/gabrielyang/finance_news_classifier-KR_v7",
-        gr.load("models/Hyeonseo/ko-finance_news_classifier",
+        gr.load("models/gabrielyang/finance_news_classifier-KR_v7",
+        # gr.load("models/Hyeonseo/ko-finance_news_classifier",
                 inputs = Link_TXT)
         Link_URL = gr.Textbox(label="뉴스 URL")