Update keyword_module.py

keyword_module.py

@@ -19,6 +19,8 @@ def summarize_kobart(text):
         truncation=True,
         max_length=512,  # 모델에 맞게 조정 (512/1024 중 하나일 확률 큼)
     )
+    if "token_type_ids" in inputs:
+        inputs.pop("token_type_ids")
 
     summary_ids = summary_model.generate(
         **inputs,