model input change

app.py

@@ -12,16 +12,16 @@ model = alexnet(pretrained=True)
 for param in model.parameters():
     param.requires_grad = False
 
-# Add a avgpool here
-avgpool = nn.AdaptiveAvgPool2d((7, 7))
-
-# Replace the classifier layer
-# to customise it according to our output
-model.classifier = nn.Sequential(
-    nn.Linear(256 * 7 * 7, 1024),
-    nn.Linear(1024, 256),
-    nn.Linear(256, 2),
-)
+# # Add a avgpool here
+# avgpool = nn.AdaptiveAvgPool2d((7, 7))
+
+# # Replace the classifier layer
+# # to customise it according to our output
+# model.classifier = nn.Sequential(
+#     nn.Linear(256 * 7 * 7, 1024),
+#     nn.Linear(1024, 256),
+#     nn.Linear(256, 2),
+# )
 
 checkpoint = torch.load(
     "CatVsDogsModel.pth", map_location=torch.device("cpu")