Update model.py

model.py

@@ -12,7 +12,7 @@ def get_model(num_classes, pretrained=True):
     model.conv1 = nn.Conv2d(1, 64, kernel_size=7, stride=2, padding=3, bias=False)
 
     # Change the output layer for our number of classes
-    model.fc = nn.
+    model.fc = nn
 Linear(model.fc.in_features, num_classes)
 
     return model