download_models.py

import subprocess

def run_command(cmd):
    print(f"🚀 Running: {cmd}")
    result = subprocess.run(cmd, shell=True)
    if result.returncode != 0:
        print(f"❌ Failed: {cmd}")
    else:
        print(f"✅ Done: {cmd}")

def main():
    # تثبيت huggingface_hub مع CLI
    run_command("pip install 'huggingface_hub[cli]'")

    # تحميل Wan2.1 I2V
    run_command("huggingface-cli download Wan-AI/Wan2.1-I2V-14B-720P --local-dir ./models/Wan2.1-I2V-14B-720P")

    # تحميل wav2vec2
    run_command("huggingface-cli download facebook/wav2vec2-base-960h --local-dir ./models/wav2vec2-base-960h")

    # تحميل FantasyTalking
    run_command("huggingface-cli download acvlab/FantasyTalking fantasytalking_model.ckpt --local-dir ./models")

if __name__ == "__main__":
    main()