diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..b0ec1bd
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,4 @@
+gender_classification_v7.csv
+gender_classification_val.csv
+gender_classification_test.csv
+gender_classification_train.csv
\ No newline at end of file
diff --git a/model.pt b/model.pt
new file mode 100644
index 0000000..c17f2d3
Binary files /dev/null and b/model.pt differ
diff --git a/train.py b/train.py
index c4dad87..ab2bbbd 100644
--- a/train.py
+++ b/train.py
@@ -11,11 +11,11 @@ import argparse
 class MyNeuralNetwork(nn.Module):
     def __init__(self, *args, **kwargs) -> None:
         super(MyNeuralNetwork, self).__init__(*args, **kwargs)
-        self.fc1 = nn.Linear(32, 7)
+        self.fc1 = nn.Linear(7, 12)
         self.relu = nn.ReLU()
-        self.fc1 = nn.Linear(12, 64)
+        self.fc1 = nn.Linear(7, 12)
         self.relu = nn.ReLU()
-        self.fc2 = nn.Linear(64, 1)
+        self.fc2 = nn.Linear(12, 1)
         self.sigmoid = nn.Sigmoid()
     
     def forward(self, x):