# requirements.txt
torch==2.0.1
torchvision==0.15.2
numpy==1.26.4
opencv-python
tqdm
gradio
timm==0.9.2
einops