Upload model.py with huggingface_hub

model.py

@@ -1,4 +1,3 @@
-
 import torch
 import torch.nn as nn
 
@@ -10,7 +9,7 @@ class SimpleCNN(nn.Module):
             nn.Conv2d(32, 64, 3, padding=1), nn.ReLU(), nn.MaxPool2d(2),
             nn.Conv2d(64, 128, 3, padding=1), nn.ReLU(), nn.AdaptiveAvgPool2d(1),
         )
-        self.fc = nn.Linear(128, 2)
+        self.fc = nn.Linear(128, 2)  # crack / no_crack
 
     def forward(self, x):
         x = self.net(x)