ndtran commited on
Commit
f692320
·
1 Parent(s): ce3e88a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ configs = {
15
  'tokenizer': 't5-small',
16
  'ignore_ids': -100,
17
  'padding_ids': 0,
18
- 'base_model': 'ndtran/t5-small_cnn-daily-mails'
19
  }
20
 
21
  model = T5ForConditionalGeneration.from_pretrained(configs['base_model'])
 
15
  'tokenizer': 't5-small',
16
  'ignore_ids': -100,
17
  'padding_ids': 0,
18
+ 'base_model': 'ndtran/t5-small_cnn-daily-mail'
19
  }
20
 
21
  model = T5ForConditionalGeneration.from_pretrained(configs['base_model'])