Hugging Face's logo

Spaces:
Overglitch
/
document-summarizer
Sleeping

App Files Community
Overglitch commited on
Commit
580f5a5
·
verified ·
1 Parent(s): 06108ff

Update modules/abstractive.py

Browse files
Files changed (1) hide show
  1. modules/abstractive.py +1 -1
modules/abstractive.py CHANGED
@@ -4,7 +4,7 @@ from transformers import PegasusTokenizer, PegasusForConditionalGeneration
4
 
5
  def load_summarizers():
6
  models = {
7
- "Pegasus": "google/pegasus-cnn_dailymail"
8
  "T5": "Overglitch/t5-small-cnn-dailymail",
9
  "BART": "facebook/bart-large-cnn",
10
  }
 
4
 
5
  def load_summarizers():
6
  models = {
7
+ "Pegasus": "google/pegasus-cnn_dailymail",
8
  "T5": "Overglitch/t5-small-cnn-dailymail",
9
  "BART": "facebook/bart-large-cnn",
10
  }