Update requirements.txt

requirements.txt

@@ -1,24 +1,37 @@
+# PyTorch - Python 3.10 compatible
 torch==2.1.0
 torchvision==0.16.0
 torchaudio==2.1.0
 numpy==1.24.4
-pydantic==2.10.6
 scipy==1.10.1
-scikit-image==0.21.0
-opencv-python==4.7.0.72
-pillow==9.4.0
+
+# IDM-VTON Core
 diffusers==0.25.0
 transformers==4.36.2
 accelerate==0.26.1
+huggingface_hub==0.25.0
+einops==0.7.0
+
+# Computer Vision
+opencv-python==4.7.0.72
+pillow==9.4.0
+scikit-image==0.21.0
+
+# UI
+gradio==4.24.0
+pydantic==2.10.6
+
+# Utilities
 matplotlib==3.7.4
 tqdm==4.64.1
 config==0.5.1
-einops==0.7.0
 onnxruntime==1.16.2
+
+# These will install latest compatible versions
 basicsr
 av
 fvcore
 cloudpickle
 omegaconf
 pycocotools
-huggingface_hub==0.25.0
+datasets