numBery commited on
Commit
482860c
1 Parent(s): 23dc51e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ def load_model():
36
 
37
  # Load T5 for Paraphrasing
38
 
39
- # t5_model = T5ForConditionalGeneration.from_pretrained('valurank/t5-paraphraser', use_auth_token=True)
40
- t5_model = export_and_get_onnx_model('valurank/t5-paraphraser')
41
  t5_tokenizer = T5Tokenizer.from_pretrained('t5-base')
42
  t5_model = t5_model.to(device)
43
  return kw_extractor, t5_model, t5_tokenizer
 
36
 
37
  # Load T5 for Paraphrasing
38
 
39
+ t5_model = T5ForConditionalGeneration.from_pretrained('valurank/t5-paraphraser', use_auth_token=True)
40
+ #t5_model = export_and_get_onnx_model('valurank/t5-paraphraser')
41
  t5_tokenizer = T5Tokenizer.from_pretrained('t5-base')
42
  t5_model = t5_model.to(device)
43
  return kw_extractor, t5_model, t5_tokenizer