Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
61 |
-
model = BigBirdForQuestionAnsweringWithNull(config,
|
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 |
|