Update evo_model.py

evo_model.py

@@ -36,9 +36,9 @@ class EvoTransformerV22(nn.Module):
         super().__init__()
         self.encoder = EvoEncoder(d_model=512, num_heads=8, ffn_dim=1024, num_layers=6, memory_enabled=True)
         self.pool = nn.AdaptiveAvgPool1d(1)
-        self.classifier = nn.Linear(512, 2)  # ✅ Matches saved model exactly
+        self.classifier = nn.Linear(512, 1)  # ✅ Matches checkpoint
 
     def forward(self, input_ids):
         x = self.encoder(input_ids)
         x = self.pool(x.transpose(1, 2)).squeeze(-1)
-        return self.classifier(x)
+        return self.classifier(x)  # Output: [batch_size, 1]