kimda commited on
Commit
1204854
โ€ข
1 Parent(s): 0dd03be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -1,18 +1,12 @@
1
  import gradio as gr
2
-
3
  from transformers import PreTrainedTokenizerFast, BartForConditionalGeneration
4
- # from transformers import๋กœ ์‹œ์ž‘ํ•˜๋Š” import ๋ฌธ์„ ๋ณด๋ฉด
5
- # ๋งŽ์€ ๊ฒฝ์šฐ AutoTokenizer, AutoModel
6
- # tokenizer = AutoTokenizer.from_pretrained("model ์ด๋ฆ„ ์–ด์ฉŒ๊ณ  ์ €์ฉŒ๊ณ ")
7
- # PreTrainedTokenizerFast : https://huggingface.co/docs/transformers/main_classes/tokenizer
8
- # BART๋Š” encoder-decoder ๋ชจ๋ธ์˜ ์˜ˆ์‹œ
9
 
10
  model_name = "ainize/kobart-news"
11
  tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name)
12
  model = BartForConditionalGeneration.from_pretrained(model_name)
13
 
14
  # ์›๋ฌธ์„ ๋ฐ›์•„์„œ ์š”์•ฝ๋ฌธ์„ ๋ฐ˜ํ™˜
15
- def summ(txt):
16
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
17
  summary_text_ids = model.generate(
18
  input_ids=input_ids,
 
1
  import gradio as gr
 
2
  from transformers import PreTrainedTokenizerFast, BartForConditionalGeneration
 
 
 
 
 
3
 
4
  model_name = "ainize/kobart-news"
5
  tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name)
6
  model = BartForConditionalGeneration.from_pretrained(model_name)
7
 
8
  # ์›๋ฌธ์„ ๋ฐ›์•„์„œ ์š”์•ฝ๋ฌธ์„ ๋ฐ˜ํ™˜
9
+ def summ(input_text): # ๋งค๊ฐœ๋ณ€์ˆ˜๋ช…์„ txt์—์„œ input_text๋กœ ๋ณ€๊ฒฝ
10
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
11
  summary_text_ids = model.generate(
12
  input_ids=input_ids,