--extra-index-url https://download.pytorch.org/whl/cu121
torch
torchvision
torchaudio

--extra-index-url https://pypi.org/simple
transformers
addict
yapf
timm
opencv-python
supervision
pycocotools
gradio