Spaces:
Sleeping
Sleeping
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)
|