Update model.py

model.py

@@ -0,0 +1,13 @@
+import torch.nn as nn
+
+class EvoTransformer(nn.Module):
+    def __init__(self, d_model=768, n_classes=2):
+        super(EvoTransformer, self).__init__()
+        self.classifier = nn.Sequential(
+            nn.Linear(d_model, 384),
+            nn.ReLU(),
+            nn.Linear(384, n_classes)
+        )
+
+    def forward(self, x):
+        return self.classifier(x)