Update README.md
Browse files
README.md
CHANGED
@@ -23,7 +23,7 @@ model = BartForConditionalGeneration.from_pretrained('gogamza/kobart-summarizati
|
|
23 |
text = "과거를 떠올려보자. 방송을 보던 우리의 모습을..."
|
24 |
|
25 |
raw_input_ids = tokenizer.encode(text)
|
26 |
-
input_ids = [tokenizer.bos_token_id] +
|
27 |
raw_input_ids + [tokenizer.eos_token_id]
|
28 |
summary_ids = model.generate(torch.tensor([input_ids]),
|
29 |
max_length=150,
|
|
|
23 |
text = "과거를 떠올려보자. 방송을 보던 우리의 모습을..."
|
24 |
|
25 |
raw_input_ids = tokenizer.encode(text)
|
26 |
+
input_ids = [tokenizer.bos_token_id] + \
|
27 |
raw_input_ids + [tokenizer.eos_token_id]
|
28 |
summary_ids = model.generate(torch.tensor([input_ids]),
|
29 |
max_length=150,
|