Update requirements.txt

requirements.txt

@@ -1,56 +1,83 @@
+# =============================================================================
 # PyTorch ecosystem (CUDA 12.1) - Must be first to avoid conflicts
+# =============================================================================
 --extra-index-url https://download.pytorch.org/whl/cu121
 torch==2.3.1
 torchvision==0.18.1
 
+# =============================================================================
 # Core scientific computing - Foundation packages
+# =============================================================================
 numpy==1.24.3
 scipy==1.11.1
 scikit-learn>=1.0.0
 py-cpuinfo
 
+# =============================================================================
 # Image and video processing
+# =============================================================================
 opencv-python-headless==4.8.1.78
 Pillow==10.1.0
 scikit-image==0.21.0
 matplotlib==3.7.2
 
+# =============================================================================
 # Video processing and codecs
+# =============================================================================
 moviepy==1.0.3
 imageio==2.31.1
 imageio-ffmpeg==0.4.9
 av==10.0.0
 
+# =============================================================================
 # Core web interface
+# =============================================================================
 gradio==4.44.0
 
+# =============================================================================
 # Enhanced SAM2 prompting dependencies - Add after PyTorch
+# =============================================================================
 ultralytics>=8.0.0
 mediapipe>=0.10.0
 
+# =============================================================================
 # SAM2 dependencies
+# =============================================================================
 onnxruntime-gpu==1.16.0
 hydra-core>=1.3.2
 omegaconf>=2.3.0
 iopath>=0.1.10
 submitit>=1.4.5
 
+# =============================================================================
 # MatAnyone dependencies - PyTorch-based packages
+# =============================================================================
 kornia>=0.7.0
 einops>=0.7.0
 timm>=0.9.2
 safetensors>=0.4.2
 
+# =============================================================================
 # Background removal
+# =============================================================================
 rembg==2.0.50
 
+# =============================================================================
 # Utilities and progress
+# =============================================================================
 tqdm==4.66.1
 requests==2.31.0
 filelock==3.12.2
 
-# HuggingFace ecosystem
+# =============================================================================
+# HuggingFace ecosystem (for AI Backgrounds / Diffusers)
+# =============================================================================
 huggingface-hub>=0.19.3
+diffusers>=0.27.2
+transformers>=4.41.2
+accelerate>=0.29.3
 
+# =============================================================================
 # System monitoring (optional)
-nvidia-ml-py3==7.352.0
+# =============================================================================
+nvidia-ml-py3==7.352.0