JordiAb commited on
Commit
fa317f2
1 Parent(s): 251abff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -61,15 +61,15 @@ Los Angeles Lakers will have more time than anticipated. The four-time NBA Most
61
  inputs = tokenizer(article_text, return_tensors='pt')
62
  # generate summary
63
  with torch.no_grad():
64
- summary_ids = model.generate(
65
- inputs['input_ids'],
66
- num_beams=4,
67
- max_length=250,
68
- early_stopping=True
69
- )
70
  # decode summary
71
  summary = tokenizer.decode(
72
- summary_ids[0],
73
- skip_special_tokens=True
74
  )
75
  ```
 
61
  inputs = tokenizer(article_text, return_tensors='pt')
62
  # generate summary
63
  with torch.no_grad():
64
+ summary_ids = model.generate(
65
+ inputs['input_ids'],
66
+ num_beams=4,
67
+ max_length=250,
68
+ early_stopping=True
69
+ )
70
  # decode summary
71
  summary = tokenizer.decode(
72
+ summary_ids[0],
73
+ skip_special_tokens=True
74
  )
75
  ```