requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.3.0+cu121     # CUDA-12 build of PyTorch 2.3
spaces                 # tiny utility lib used by Hugging Face Spaces