Update app.py

app.py

@@ -42,10 +42,11 @@ class GenreCNN(nn.Module):
             nn.Conv2d(128, 256, 3, padding=1), nn.BatchNorm2d(256),
             nn.ReLU(), nn.MaxPool2d(2, 2), nn.Dropout2d(0.25)
         )
-        self.pool       = nn.AdaptiveAvgPool2d((1, 1))
+        self.pool       = nn.AdaptiveAvgPool2d((4, 4))   # 256*4*4 = 4096
         self.classifier = nn.Sequential(
             nn.Flatten(),
-            nn.Linear(256, 128), nn.ReLU(), nn.Dropout(0.5),
+            nn.Linear(4096, 512), nn.ReLU(), nn.Dropout(0.5),
+            nn.Linear(512, 128),  nn.ReLU(), nn.Dropout(0.5),
             nn.Linear(128, n_classes)
         )