Leesn465 commited on
Commit
536adce
·
verified ·
1 Parent(s): cefe789

Update util/keywordExtract.py

Browse files
Files changed (1) hide show
  1. util/keywordExtract.py +3 -2
util/keywordExtract.py CHANGED
@@ -18,8 +18,9 @@ def load_company_list(file_path='상장법인목록.xls'):
18
  return df['회사명'].dropna().tolist()
19
 
20
  # ✅ 요약용 KoBART
21
- summary_tokenizer = PreTrainedTokenizerFast.from_pretrained("gogamza/kobart-summarization")
22
- summary_model = BartForConditionalGeneration.from_pretrained("gogamza/kobart-summarization")
 
23
 
24
  def summarize_kobart(text):
25
  input_ids = summary_tokenizer.encode(text, return_tensors="pt")
 
18
  return df['회사명'].dropna().tolist()
19
 
20
  # ✅ 요약용 KoBART
21
+ SUMMARY_MODEL_ID = "ainize/kobart-news" # 👈 안정적인 KoBART 모델 ID 사용
22
+ summary_tokenizer = PreTrainedTokenizerFast.from_pretrained(SUMMARY_MODEL_ID)
23
+ summary_model = BartForConditionalGeneration.from_pretrained(SUMMARY_MODEL_ID)
24
 
25
  def summarize_kobart(text):
26
  input_ids = summary_tokenizer.encode(text, return_tensors="pt")