Update evo_model.py

evo_model.py

@@ -36,11 +36,7 @@ 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.Sequential(
-            nn.Linear(512, 128),
-            nn.ReLU(),
-            nn.Linear(128, 2)
-        )
+        self.classifier = nn.Linear(512, 2)  # ✅ Matches saved model exactly
 
     def forward(self, input_ids):
         x = self.encoder(input_ids)