czq commited on
Commit
2d432c9
·
1 Parent(s): aebe725

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,8 +57,8 @@ class BigBirdForQuestionAnsweringWithNull(PreTrainedModel):
57
 
58
  return (outputs.start_logits, outputs.end_logits, null_logits)
59
  model_id = 'vasudevgupta/bigbird-roberta-natural-questions'
60
- config = BigBirdConfig.from_pretrained(model_path)
61
- model = BigBirdForQuestionAnsweringWithNull(config, model_path)
62
  # model.to('cuda')
63
  model.eval()
64
 
 
57
 
58
  return (outputs.start_logits, outputs.end_logits, null_logits)
59
  model_id = 'vasudevgupta/bigbird-roberta-natural-questions'
60
+ config = BigBirdConfig.from_pretrained(model_id)
61
+ model = BigBirdForQuestionAnsweringWithNull(config, model_id)
62
  # model.to('cuda')
63
  model.eval()
64