nppmatt commited on
Commit
70d0f55
1 Parent(s): 2206dc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ class PretrainedBertClass(torch.nn.Module):
94
 
95
  # return_dict must equal False for Huggingface Transformers v4+
96
  def forward(self, input_ids, attention_mask, token_type_ids):
97
- output = self.l1(
98
  input_ids=input_ids,
99
  attention_mask=attention_mask,
100
  token_type_ids=token_type_ids,
 
94
 
95
  # return_dict must equal False for Huggingface Transformers v4+
96
  def forward(self, input_ids, attention_mask, token_type_ids):
97
+ _, output = self.l1(
98
  input_ids=input_ids,
99
  attention_mask=attention_mask,
100
  token_type_ids=token_type_ids,