isLinXu commited on
Commit
41edb9e
·
1 Parent(s): 0e66889
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -53,6 +53,10 @@ def predict_pose(img, model_name, out_dir):
53
  result = next(result_generator)
54
  save_dir = './output/visualizations/'
55
  out_img_path = save_dir + img_path
 
 
 
 
56
  out_img = PIL.Image.open(out_img_path)
57
  return out_img
58
 
 
53
  result = next(result_generator)
54
  save_dir = './output/visualizations/'
55
  out_img_path = save_dir + img_path
56
+ files = os.listdir(out_dir)
57
+ for file in files:
58
+ print("file:", file)
59
+ print("out_img_path:", out_img_path)
60
  out_img = PIL.Image.open(out_img_path)
61
  return out_img
62