face expressions

app/Hackathon_setup/exp_recognition.py

@@ -60,8 +60,14 @@ def get_expression(img):
     # Recreate the same model architecture
     num_classes = 7  # 👈 change this to match your training setup
 
-    model = models.resnet18(weights=None)  # no pretrained weights now
-    model.fc = nn.Linear(model.fc.in_features, num_classes)
+    model = models.resnet18(weights=None)
+    model.conv1 = nn.Conv2d(1, 64, kernel_size=7, stride=2, padding=3, bias=False)
+    # no pretrained weights now
+    model.fc = nn.Sequential(
+        nn.Linear(model.fc.in_features, 256),
+        nn.ReLU(inplace=True),
+        nn.Linear(256, num_classes)
+    )
 
     model = model.to(device)