requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.1.1
torchvision==0.16.1
transformers>=4.36.0
gradio==4.7.1
Pillow==10.1.0
numpy==1.24.3
datasets==2.14.5
tqdm==4.66.1
einops==0.6.1
timm==0.9.2
pandas==2.0.3
scikit-learn==1.3.2