Update app.py

app.py

@@ -131,13 +131,9 @@ mp_face_mesh = mp.solutions.face_mesh
 face_detection = mp_face_detection.FaceDetection(model_selection=1, min_detection_confidence=0.5)
 face_mesh = mp_face_mesh.FaceMesh(static_image_mode=False, max_num_faces=1, min_detection_confidence=0.5)
 
-# Initialize ResNet-34 model with random weights
-def create_model():
-    model = models.resnet34(pretrained=False)
-    model.fc = torch.nn.Linear(model.fc.in_features, 2)
-    return model
-
-model = create_model()
+# Load the model from the .pkl file
+model = create_model()  # Ensure architecture matches the saved model
+model.load_state_dict(torch.load("resnet34.pkl", map_location=torch.device('cpu')))  # Use 'cuda' if loading on GPU
 model.eval()
 
 # Define transformation for face images