# main/requirements.txt

# --- Core Dependencies for the IDM-VTON Model ---
# Pinning versions that are known to be stable together
torch==2.1.2
torchvision==0.16.2
transformers==4.36.2
diffusers==0.25.0
accelerate==0.26.1
huggingface_hub==0.20.3

# --- Dependencies for DensePose & Detectron2 ---
# These are required by the local detectron2/densepose libraries
fvcore
pyyaml==5.1
# This command tells pip to install the 'detectron2' library from the local directory.
# The '-e' flag installs it in "editable" mode, which is standard for local packages.
-e .

# --- Image and Data Processing Libraries ---
numpy==1.24.4
Pillow==9.4.0
scikit-image==0.21.0
opencv-python==4.7.0.72
scipy==1.10.1

# --- Other Utilities ---
gradio
matplotlib==3.7.4
tqdm==4.64.1
config==0.5.1
einops==0.7.0
onnxruntime==1.16.2
basicsr
av
cloudpickle
omegaconf
pycocotools