Leesn465 commited on
Commit
b7386aa
·
verified ·
1 Parent(s): 6db7357

Update util/keywordExtract.py

Browse files
Files changed (1) hide show
  1. util/keywordExtract.py +3 -3
util/keywordExtract.py CHANGED
@@ -18,9 +18,9 @@ def load_company_list(file_path='상장법인목록.xls'):
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")
 
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
 
25
  def summarize_kobart(text):
26
  input_ids = summary_tokenizer.encode(text, return_tensors="pt")