Update util/keywordExtract.py

util/keywordExtract.py

@@ -22,15 +22,22 @@ summary_tokenizer = PreTrainedTokenizerFast.from_pretrained("gogamza/kobart-summ
 summary_model = BartForConditionalGeneration.from_pretrained("gogamza/kobart-summarization")
 
 def summarize_kobart(text):
-    input_ids = summary_tokenizer.encode(text, return_tensors="pt")
+    # ✅ 입력 길이 제한(핵심)
+    inputs = summary_tokenizer(
+        text,
+        return_tensors="pt",
+        truncation=True,
+        max_length=1024,  # 모델에 맞게 조정 (512/1024 중 하나일 확률 큼)
+    )
+
     summary_ids = summary_model.generate(
-        input_ids,
-        max_length=160,
-        min_length=100,
+        **inputs,
+        max_new_tokens=160,   # ✅ 출력 길이는 max_new_tokens로 관리 추천
+        min_new_tokens=100,
         num_beams=4,
         repetition_penalty=2.5,
         no_repeat_ngram_size=4,
-        early_stopping=True
+        early_stopping=True,
     )
     return summary_tokenizer.decode(summary_ids[0], skip_special_tokens=True)