Update app.py

app.py

@@ -17,14 +17,14 @@ class EmotionModel(nn.Module):
     def __init__(self):
         super(EmotionModel, self).__init__()
         self.model = nn.Sequential(
-            nn.Conv2d(1, 16, kernel_size=3, stride=1, padding=1),
+            nn.Conv2d(1, 32, 3, padding=1),   # Match 32 out_channels
             nn.ReLU(),
             nn.MaxPool2d(2, 2),
-            nn.Conv2d(16, 32, kernel_size=3, stride=1, padding=1),
+            nn.Conv2d(32, 64, 3, padding=1),  # Match 64 out_channels
             nn.ReLU(),
             nn.MaxPool2d(2, 2),
             nn.Flatten(),
-            nn.Linear(32 * 12 * 12, 128),
+            nn.Linear(64 * 12 * 12, 128),     # 64 filters * 12x12 features = 9216
             nn.ReLU(),
             nn.Linear(128, 7)
         )