torch==2.1.2+cu121
torchvision==0.16.2+cu121
torchaudio==2.1.2+cu121
--extra-index-url https://download.pytorch.org/whl/cu121
tqdm
gradio==4.31.5
pydantic==2.5.0
pillow