control-animation / utils /gradio_utils.py
Pie31415's picture
initial commit
2c924d3
raw
history blame
721 Bytes
import os
# App Pose utils
def motion_to_video_path(motion):
videos = [
"__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