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