requirements.txt

--extra-index-url https://download.pytorch.org/whl/cpu

gradio ~=4.36.0
torch >=2.0.0
torchvision >=0.15.0
pytorch-lightning ~=2.2.0
timm ~=0.9.16
nltk