Gladiator commited on
Commit
e7fc023
1 Parent(s): 32ff21e

bart large cnn

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,8 +30,8 @@ if __name__ == "__main__":
30
  # ---------------------------
31
  # SETUP & Constants
32
  nltk.download("punkt")
33
- abs_tokenizer_name = "t5-base"
34
- abs_model_name = "t5-base"
35
  abs_tokenizer = T5Tokenizer.from_pretrained(abs_tokenizer_name)
36
  abs_max_length = 80
37
  abs_min_length = 30
 
30
  # ---------------------------
31
  # SETUP & Constants
32
  nltk.download("punkt")
33
+ abs_tokenizer_name = "facebook/bart-large-cnn"
34
+ abs_model_name = "facebook/bart-large-cnn"
35
  abs_tokenizer = T5Tokenizer.from_pretrained(abs_tokenizer_name)
36
  abs_max_length = 80
37
  abs_min_length = 30