Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from transformers import PreTrainedTokenizerFast, BartForConditionalGeneration
|
|
8 |
model_name = "ainize/kobart-news"
|
9 |
tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name)
|
10 |
model = BartForConditionalGeneration.from_pretrained(model_name)
|
11 |
-
|
12 |
input_ids=tokenizer.encode(txt, return_tensors="pt")
|
13 |
summary_text_ids=model.generate(
|
14 |
input_ids=input_ids,
|
|
|
8 |
model_name = "ainize/kobart-news"
|
9 |
tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name)
|
10 |
model = BartForConditionalGeneration.from_pretrained(model_name)
|
11 |
+
def summ(txt):
|
12 |
input_ids=tokenizer.encode(txt, return_tensors="pt")
|
13 |
summary_text_ids=model.generate(
|
14 |
input_ids=input_ids,
|