gogamza commited on
Commit
f5bdbaa
1 Parent(s): 23dfd9c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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,