Update requirements.txt

requirements.txt

@@ -1,24 +1,34 @@
-torch==2.1.1
-torchvision==0.16.1
-diffusers==0.31.0
-transformers==4.46.3
-accelerate==1.1.1
-gradio==5.49.1
-pillow==11.0.0
-numpy==1.26.4
-opencv-python==4.10.0.84
-safetensors==0.4.5
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+# Core PyTorch Ecosystem
+torch==2.8.0
+torchvision==0.23.0
+xformers==0.0.32.post2
+
+# Hugging Face Ecosystem
+transformers==4.57.0
+diffusers==0.34.0
+accelerate==1.11.0
+huggingface-hub
+peft==0.17.1
+
+# ML/Computer Vision Libraries
+onnxruntime==1.23.2
+onnx==1.19.1
 insightface==0.7.3
-onnxruntime-gpu==1.19.2
-onnx==1.17.0
-scikit-image==0.24.0
-scipy==1.14.1
+mediapipe==0.10.21
+controlnet-aux==0.0.10
+opencv-python-headless
+Pillow==11.3
+scikit-image==0.25.2
+
+# Scientific Python Stack
+numpy==1.24.0
+scipy==1.15.3
+safetensors==0.6.2
+
+# Utilities
 omegaconf==2.3.0
-einops==0.8.0
-huggingface-hub==0.33.5
-peft==0.13.2
-xformers
-spaces
-controlnet-aux  # NEW: For ZoeDetector (better depth estimation)
-compel  # NEW: For better prompt handling (optional but recommended)
-mediapipe # NEW: Needed in new update
+einops==0.8.1
+gradio==5.49.0
+compel==2.0.3