utils.py

def install_packages():
    import subprocess
    import sys
    import importlib

    required = [
        'torch>=2.4.0,<2.6.0',
        'diffusers',
        'transformers',
        'gradio',
        'safetensors',
        'huggingface_hub',
        'imageio',
        'opencv-python-headless',
        'Pillow'
    ]

    subprocess.run([
        sys.executable, "-m", "pip", "install",
        "--upgrade", "pip", "setuptools", "wheel"
    ], check=True)

    subprocess.run([
        sys.executable, "-m", "pip", "install"
    ] + required, check=True)