update

app.py

@@ -51,9 +51,12 @@ def predict_pose(img, model_name, out_dir):
     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/'
-    out_img_path = save_dir + img_path
-    print("out_img_path: ", out_img_path)
+    save_dir = './output/visualizations/'
+    if os.path.exists(save_dir):
+        out_img_path = save_dir + img_path
+        print("out_img_path: ", out_img_path)
+    else:
+        out_img_path = img_path
     out_img = PIL.Image.open(out_img_path)
     return out_img