onebeans commited on
Commit
4d7bcf1
ยท
verified ยท
1 Parent(s): 38c216c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -56,11 +56,11 @@ def load_summarization_model(model_dir):
56
  return None, None
57
 
58
  # ๋ชจ๋ธ ๊ฒฝ๋กœ ์„ค์ • (์‚ฌ์šฉ์ž ํ™˜๊ฒฝ์— ๋งž๊ฒŒ ๋ณ€๊ฒฝ)
59
- checkpoint_path = 'onebeans/keT5-news-summarizer' # ํŒŒ์ธํŠœ๋‹๋œ T5 ์ œ๋ชฉ ์ƒ์„ฑ ๋ชจ๋ธ ๊ฒฝ๋กœ
60
- model_dir = 'onebeans/keT5-news-title-gen' # ๋‰ด์Šค ๊ธฐ์‚ฌ ์š”์•ฝ ๋ชจ๋ธ ๊ฒฝ๋กœ
61
 
62
- title_model, title_tokenizer = load_title_model(checkpoint_path)
63
- summarizer_model, summarizer_tokenizer = load_summarization_model(model_dir)
64
 
65
  # ๋‰ด์Šค ๊ธฐ์‚ฌ ์ œ๋ชฉ ์ƒ์„ฑ ํ•จ์ˆ˜
66
  def generate_title(article_text, max_length=20, num_beams=10, early_stopping=True):
 
56
  return None, None
57
 
58
  # ๋ชจ๋ธ ๊ฒฝ๋กœ ์„ค์ • (์‚ฌ์šฉ์ž ํ™˜๊ฒฝ์— ๋งž๊ฒŒ ๋ณ€๊ฒฝ)
59
+ title_path = 'onebeans/keT5-news-title-gen' # ํŒŒ์ธํŠœ๋‹๋œ T5 ์ œ๋ชฉ ์ƒ์„ฑ ๋ชจ๋ธ ๊ฒฝ๋กœ
60
+ abs_path = 'onebeans/keT5-news-summarizer' # ๋‰ด์Šค ๊ธฐ์‚ฌ ์š”์•ฝ ๋ชจ๋ธ ๊ฒฝ๋กœ
61
 
62
+ title_model, title_tokenizer = load_title_model(title_path)
63
+ summarizer_model, summarizer_tokenizer = load_summarization_model(abs_path)
64
 
65
  # ๋‰ด์Šค ๊ธฐ์‚ฌ ์ œ๋ชฉ ์ƒ์„ฑ ํ•จ์ˆ˜
66
  def generate_title(article_text, max_length=20, num_beams=10, early_stopping=True):