import os

def create_directory(path):
    if not os.path.exists(path):
        os.makedirs(path)
    
def download():
    emoji_path = './ckpt_models/'
    create_directory(emoji_path)          
    os.system(f'git clone https://code.openxlab.org.cn/houshaowei/FollowYourEmoji.git {emoji_path}')
    os.system(f'cd {emoji_path} && git lfs pull')   
    
    
    base_path = './ckpt_models/base'
    create_directory(base_path)          
    os.system(f'git clone https://code.openxlab.org.cn/houshaowei/FollowYourEmoji_BaseModelPack.git {base_path}')
    os.system(f'cd {base_path} && git lfs pull')   

if __name__ == "__main__":
    download()