quantumiracle-git commited on
Commit
21286a0
·
1 Parent(s): 7eab692

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -105,6 +105,7 @@ def update(user_choice, left, right, data_folder=VIDEO_PATH, flag_to_huggingface
105
  if f.endswith(f'.{FORMAT}'):
106
  video_files.append(os.path.join(data_folder, env_name, f))
107
  # randomly choose two videos
 
108
  selected_video_ids = np.random.choice(len(video_files), 2, replace=False)
109
  left = video_files[selected_video_ids[0]]
110
  right = video_files[selected_video_ids[1]]
@@ -117,6 +118,7 @@ def replay(left, right):
117
 
118
  def parse_envs(folder=VIDEO_PATH):
119
  envs = []
 
120
  for f in os.listdir(folder):
121
  if os.path.isdir(os.path.join(folder, f)):
122
  envs.append(f)
 
105
  if f.endswith(f'.{FORMAT}'):
106
  video_files.append(os.path.join(data_folder, env_name, f))
107
  # randomly choose two videos
108
+ print(video_files, envs)
109
  selected_video_ids = np.random.choice(len(video_files), 2, replace=False)
110
  left = video_files[selected_video_ids[0]]
111
  right = video_files[selected_video_ids[1]]
 
118
 
119
  def parse_envs(folder=VIDEO_PATH):
120
  envs = []
121
+ print('folder: ', os.listdir(folder))
122
  for f in os.listdir(folder):
123
  if os.path.isdir(os.path.join(folder, f)):
124
  envs.append(f)