Update model.py

model.py

@@ -138,6 +138,7 @@ class Model:
         self.device = torch.device(
             'cuda:0' if torch.cuda.is_available() else 'cpu')    
         self.model_dir = pathlib.Path(model_dir)
+        self.model_dir.mkdir(exist_ok=True, parents=True)
         self.download_pose_models()
         self.download_models()
 
@@ -147,14 +148,14 @@ class Model:
         device = "cuda"
         det_config_file = model_path+"faster_rcnn_r50_fpn_coco.py"
         subprocess.run(shlex.split(f'wget {det_config_file} -O models/faster_rcnn_r50_fpn_coco.py'))
-        det_config = 'models/faster_rcnn_r50_fpn_coco.py' 
+        det_config = 'models/faster_rcnn_r50_fpn_coco.py'
         
         det_checkpoint_file = "https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth"
         subprocess.run(shlex.split(f'wget {det_checkpoint_file} -O models/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'))
         det_checkpoint = 'models/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'
 
         pose_config_file = model_path+"hrnet_w48_coco_256x192.py"
-        subprocess.run(shlex.split(f'wget {pose_config_file} -O models/rnet_w48_coco_256x192.py'))
+        subprocess.run(shlex.split(f'wget {pose_config_file} -O models/hrnet_w48_coco_256x192.py'))
         pose_config = 'models/hrnet_w48_coco_256x192.py'
 
         pose_checkpoint_file = "https://download.openmmlab.com/mmpose/top_down/hrnet/hrnet_w48_coco_256x192-b9e0b3ab_20200708.pth"
@@ -170,7 +171,6 @@ class Model:
         self.pose_model = init_pose_model(pose_config_mmcv, pose_checkpoint, device=device)
 
     def download_models(self) -> None:
-        self.model_dir.mkdir(exist_ok=True, parents=True)
         device = 'cuda'
     
         config = OmegaConf.load("configs/stable-diffusion/test_sketch.yaml")