Spaces:
Build error
Build error
File size: 854 Bytes
2c924d3 71e9a42 2c924d3 71e9a42 2c924d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import os
# App Pose utils
def motion_to_video_path(motion):
videos = [
"__assets__/walk_01.mp4",
"__assets__/walk_02.mp4",
"__assets__/walk_03.mp4",
"__assets__/run.mp4",
"__assets__/dance1_corr.mp4",
"__assets__/dance2_corr.mp4",
"__assets__/dance3_corr.mp4",
"__assets__/dance4_corr.mp4",
"__assets__/dance5_corr.mp4",
]
if len(motion.split(" ")) > 1 and motion.split(" ")[1].isnumeric():
id = int(motion.split(" ")[1]) - 1
return videos[id]
else:
return motion
def logo_name_to_path(name):
logo_paths = {
'Picsart AI Research': '__assets__/pair_watermark.png',
'Text2Video-Zero': '__assets__/t2v-z_watermark.png',
'None': None
}
if name in logo_paths:
return logo_paths[name]
return name
|