Update model.py

model.py

@@ -51,5 +51,5 @@ class FleshkaTabularTransformer(PreTrainedModel):
             x = x.unsqueeze(1)               # [batch_size, 1, d_model] (добавляем seq_len=1)
             x = self.transformer(x)          # [batch_size, 1, d_model]
             x = x.squeeze(1)                 # [batch_size, d_model]
-            out.append(self.head(x).item().item() > 0)
+            out.append(self.head(x).item() > 0)
         return out