quantumiracle-git commited on
Commit
699ef40
·
1 Parent(s): 21286a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,6 @@ 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
- 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]]
@@ -148,6 +147,7 @@ def build_interface(iter=3, data_folder=VIDEO_PATH):
148
  if f.endswith(f'.{FORMAT}'):
149
  video_files.append(os.path.join(data_folder, env_name, f))
150
  # randomly choose two videos
 
151
  selected_video_ids = np.random.choice(len(video_files), 2, replace=False)
152
  left_video_path = video_files[selected_video_ids[0]]
153
  right_video_path = video_files[selected_video_ids[1]]
 
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]]
 
147
  if f.endswith(f'.{FORMAT}'):
148
  video_files.append(os.path.join(data_folder, env_name, f))
149
  # randomly choose two videos
150
+ print(video_files, envs)
151
  selected_video_ids = np.random.choice(len(video_files), 2, replace=False)
152
  left_video_path = video_files[selected_video_ids[0]]
153
  right_video_path = video_files[selected_video_ids[1]]