requirements.txt

# Core dependencies
torch>=2.1.0
torchvision>=0.16.0
gradio==3.50.2
pillow==9.4.0
opencv-python-headless==4.8.1.78
pyyaml==6.0.1
json_tricks==3.17.3
yacs==0.1.8
scikit-learn==1.3.1
pandas==2.0.3
timm==0.4.12
numpy==1.26.4
einops==0.8.0
fvcore==0.1.5.post20221221

# Model and inference
transformers==4.34.0
sentencepiece==0.1.99
ftfy==6.1.1
regex==2023.10.3
nltk==3.8.1
vision-datasets==0.2.2
pycocotools==2.0.7
scikit-image==0.21.0
accelerate==0.23.0
kornia==0.7.0
open-clip-torch==2.26.1
huggingface-hub

# Build dependencies
cython==3.0.2
setuptools>=59.5.0
mpi4py==3.1.5

# Detectron2
git+https://github.com/MaureenZOU/detectron2-xyz.git

# Visualization
matplotlib