requirements.txt

gradio
--index-url https://download.pytorch.org/whl/cu121
torch>=2.1.1
torchvision 
torchaudio