Update landmark_detection.py

landmark_detection.py

@@ -56,8 +56,8 @@ def load_model(weights_path):
             local_dir=".",
         )
     checkpoint = torch.load(weights_path, map_location=device)
-    model.load_state_dict(checkpoint)
-    # model.load_state_dict(checkpoint["state_dict_backbone"])
+    # model.load_state_dict(checkpoint)
+    model.load_state_dict(checkpoint["state_dict_backbone"])
     model = model.eval()
     model = model.to(dtype=dtype)
     # model = torch.compile(model, mode="reduce-overhead")