Commit
·
699ef40
1
Parent(s):
21286a0
Update app.py
Browse files
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]]
|