update

app.py

@@ -46,15 +46,15 @@ def download_test_image():
 
 def predict_pose(img, model_name, out_dir):
     out_dir = "./output/visualizations/"
-    if not os.path.exists(out_dir):
-        os.makedirs(out_dir)
     img_path = "input_img.jpg"
     save_image(img, img_path)
     device = torch.cuda.current_device() if torch.cuda.is_available() else 'cpu'
     inferencer = MMPoseInferencer(model_name, device=device)
     result_generator = inferencer(img_path, show=False, out_dir=out_dir)
     result = next(result_generator)
-    save_dir = './output/visualizations/'
+    save_dir = './output/output/visualizations/'
+    if not os.path.exists(out_dir):
+        os.makedirs(out_dir)
     out_img_path = save_dir + img_path
     print("out_img_path: ", out_img_path)
     out_img = PIL.Image.open(out_img_path)