Théo Rousseaux commited on
Commit
9ad7a80
·
1 Parent(s): dbaecc5

correction path

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -96,8 +96,9 @@ with col2:
96
  predict_list = os.listdir(os.path.join('/home/user/.pyenv/runs', 'pose'))
97
  predict_list.sort()
98
  predict_dir = predict_list[-1]
99
- print(os.listdir(os.path.join('/home/user/.pyenv/runs', 'pose', predict_dir)))
100
- st.video(os.path.join('/home/user/.pyenv/runs', 'pose', predict_dir, 'squat.mp4'), loop=True)
 
101
  else :
102
  st.video(video_uploaded)
103
 
 
96
  predict_list = os.listdir(os.path.join('/home/user/.pyenv/runs', 'pose'))
97
  predict_list.sort()
98
  predict_dir = predict_list[-1]
99
+ file_name = os.listdir(os.path.join('/home/user/.pyenv/runs', 'pose', predict_dir))[0]
100
+ print(file_name)
101
+ st.video(os.path.join('/home/user/.pyenv/runs', 'pose', predict_dir, file_name), loop=True)
102
  else :
103
  st.video(video_uploaded)
104