Hugging Face's logo

Spaces:
JikeStardy
/
Sum4rize
Paused

App Files Community
tomato commited on
Commit
e6eb31e
•
1 Parent(s): c872f8c

use another model [csebuetnlp/mT5_multilingual_XLSum]

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from tqdm import tqdm
4
  from transformers import PegasusForConditionalGeneration
5
  from tokenizers_pegasus import PegasusTokenizer
6
 
7
- MODEL_NAME = "IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese"
8
 
9
  summarizer = PegasusForConditionalGeneration.from_pretrained(MODEL_NAME)
10
  tokenizer = PegasusTokenizer.from_pretrained(MODEL_NAME)
 
4
  from transformers import PegasusForConditionalGeneration
5
  from tokenizers_pegasus import PegasusTokenizer
6
 
7
+ MODEL_NAME = "csebuetnlp/mT5_multilingual_XLSum"
8
 
9
  summarizer = PegasusForConditionalGeneration.from_pretrained(MODEL_NAME)
10
  tokenizer = PegasusTokenizer.from_pretrained(MODEL_NAME)