Théo Rousseaux commited on
Commit
a586a8c
1 Parent(s): 4c8d193

hugging face path

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -87,12 +87,12 @@ with col2:
87
  video_uploaded = save_uploaded_file(video_uploaded)
88
  if video_uploaded.split("/")[-1] != st.session_state.file_name:
89
  shutil.rmtree('fig', ignore_errors=True)
90
- shutil.rmtree('runs', ignore_errors=True)
91
  st.session_state.file_name = None
92
  st.session_state.file_name = video_uploaded.split("/")[-1]
93
- _left, mid, _right = st.columns(3)
94
  with mid:
95
- if os.path.exists('runs'):
96
  predict_list = os.listdir(os.path.join('runs', 'pose'))
97
  predict_list.sort()
98
  predict_dir = predict_list[-1]
 
87
  video_uploaded = save_uploaded_file(video_uploaded)
88
  if video_uploaded.split("/")[-1] != st.session_state.file_name:
89
  shutil.rmtree('fig', ignore_errors=True)
90
+ shutil.rmtree('/home/user/.pyenv/runs', ignore_errors=True)
91
  st.session_state.file_name = None
92
  st.session_state.file_name = video_uploaded.split("/")[-1]
93
+ _left, mid, _right = st.columns([1, 3, 1])
94
  with mid:
95
+ if os.path.exists('/home/user/.pyenv/runs'):
96
  predict_list = os.listdir(os.path.join('runs', 'pose'))
97
  predict_list.sort()
98
  predict_dir = predict_list[-1]