Update app.py

app.py

@@ -94,7 +94,7 @@ class PretrainedBertClass(torch.nn.Module):
 
     # return_dict must equal False for Huggingface Transformers v4+
     def forward(self, input_ids, attention_mask, token_type_ids):
-        output = self.l1(
+        _, output = self.l1(
             input_ids=input_ids,
             attention_mask=attention_mask,
             token_type_ids=token_type_ids,