Update landmark_detection.py

landmark_detection.py

@@ -49,13 +49,11 @@ transforms_image = torchvision.transforms.Compose(
 def load_model(weights_path):
     model = FaceXFormer().to(device)
     if not os.path.exists(weights_path):
-        name = os.path.basename(weights_path)
-        path = os.path.dirname(weights_path)
         huggingface_hub.hf_hub_download(
             "kartiknarayan/facexformer",
             "ckpts/model.pt",
             repo_type="model",
-            local_dir=path,
+            local_dir=".",
         )
     checkpoint = torch.load(weights_path, map_location=device)
     model.load_state_dict(checkpoint)