Fix model to receive input_ids

model.py

@@ -52,7 +52,7 @@ class BertMesh(PreTrainedModel):
             outs = torch.sigmoid(self.linear_2(outs))
             outs = torch.flatten(outs, start_dim=1)
         else:
-            cls = self.bert(input_ids=inputs)[1]
+            cls = self.bert(input_ids=input_ids)[1]
             outs = torch.nn.functional.relu(self.linear_1(cls))
             outs = self.dropout_layer(outs)
             outs = torch.sigmoid(self.linear_out(outs))