toeknee432 commited on
Commit
a2a8bda
·
1 Parent(s): 3d7767b

Update pages/3_Abstractive Summary.py

Browse files
Files changed (1) hide show
  1. pages/3_Abstractive Summary.py +1 -0
pages/3_Abstractive Summary.py CHANGED
@@ -17,6 +17,7 @@ def load_model():
17
  return model,tokenizer
18
 
19
  model,tokenizer = load_model()
 
20
  #run this the first time and use the local model for faster runtime
21
  tokenizer.save_pretrained("local_pegasus-large_tokenizer")
22
  model.save_pretrained("local_pegasus-large_tokenizer_model")
 
17
  return model,tokenizer
18
 
19
  model,tokenizer = load_model()
20
+
21
  #run this the first time and use the local model for faster runtime
22
  tokenizer.save_pretrained("local_pegasus-large_tokenizer")
23
  model.save_pretrained("local_pegasus-large_tokenizer_model")