Update requirements.txt

requirements.txt

@@ -1,31 +1,38 @@
 # main/requirements.txt
 
-# --- Core ML/DL Libraries ---
-torch
-torchvision
-torchaudio
-transformers==4.38.2
-diffusers==0.27.2 # Upgraded from 0.25.0
-accelerate==0.28.0 # Upgraded from 0.26.1
-huggingface_hub # Un-pinning the version to let diffusers choose
+# --- 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
 
-# --- Data & Image Processing ---
-numpy==1.26.4
-scipy==1.11.4
-scikit-image==0.22.0
-opencv-python==4.8.0.76
-Pillow==10.1.0
+# --- 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
 
-# --- Utilities & Others ---
-matplotlib==3.8.2
-tqdm==4.66.1
+# --- Other Utilities ---
+gradio
+matplotlib==3.7.4
+tqdm==4.64.1
+config==0.5.1
 einops==0.7.0
-omegaconf
-# The following probably don't need version pins
-config
-onnxruntime
+onnxruntime==1.16.2
 basicsr
 av
-fvcore
 cloudpickle
+omegaconf
 pycocotools