Update requirements.txt

requirements.txt

@@ -24,23 +24,17 @@ python-multipart>=0.0.6,<1.0
 # ============================================================================
 # DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1)
 # ============================================================================
-# Latest stable PyTorch for best performance
 torch==2.5.1
 torchvision==0.20.1
 torchaudio==2.5.1
-
-# Hugging Face ecosystem - Latest stable for SAM2 support
 transformers>=4.45.0,<5.0
 huggingface_hub>=0.25.0,<1.0
 accelerate>=0.20.3,<1.0
 safetensors==0.4.3
 tokenizers>=0.19.0,<1.0
 datasets>=2.14.0,<3.0
-
-# Model utilities
 einops==0.8.0
 timm>=0.9.16,<1.0
-# Loosened to allow a version compatible with torch 2.5.1
 xformers>=0.0.28,<0.1.0
 
 # ============================================================================
@@ -48,17 +42,13 @@ xformers>=0.0.28,<0.1.0
 # ============================================================================
 opencv-python-headless==4.10.0.84
 opencv-contrib-python-headless==4.10.0.84
-
-# Video processing
 moviepy>=1.0.3,<2.0
-# Allow any imageio ≥2.25.0 (to satisfy MatAnyone) up through 2.34.x
-imageio>=2.25.0,<2.35.0
+# imageio must satisfy scikit-image:
+imageio>=2.33.0,<2.35.0
 imageio-ffmpeg>=0.4.9,<1.0
 ffmpeg-python>=0.2.0,<1.0
-av>=11.0.0,<12.0  # PyAV for better video handling
-
-# Image processing
-scikit-image>=0.22.0,<1.0
+av>=11.0.0,<12.0
+scikit-image>=0.33.0,<1.0
 albumentations>=1.3.1,<2.0
 
 # ============================================================================
@@ -84,19 +74,19 @@ typer>=0.9.0,<1.0
 # ============================================================================
 # PERFORMANCE & OPTIMIZATION
 # ============================================================================
-numba>=0.58.0,<1.0  # JIT compilation for speed
-joblib>=1.3.2,<2.0  # Parallel processing
+numba>=0.58.0,<1.0
+joblib>=1.3.2,<2.0
 diskcache>=5.6.3,<6.0
 psutil>=5.9.5,<6.0
-gputil>=1.4.0,<2.0  # GPU monitoring
-pynvml>=11.5.0,<12.0  # NVIDIA management
+gputil>=1.4.0,<2.0
+pynvml>=11.5.0,<12.0
 
 # ============================================================================
 # LOGGING & MONITORING
 # ============================================================================
 loguru>=0.7.2,<1.0
 tensorboard>=2.15.0,<3.0
-wandb>=0.16.0,<1.0  # Experiment tracking
+wandb>=0.16.0,<1.0
 prometheus-client>=0.19.0,<1.0
 
 # ============================================================================
@@ -107,16 +97,10 @@ marshmallow>=3.20.0,<4.0
 cerberus>=1.3.5,<2.0
 
 # ============================================================================
-# MATANYONE DEPENDENCIES (Python 3.10 compatible)
+# MATANYONE (install separately, no-deps)
 # ============================================================================
-easydict==1.13  # Updated version
-gdown>=4.7.1,<5.0
-hickle>=5.0.2
-cchardet>=2.1.7
-gitpython>=3.1.40,<4.0
-netifaces>=0.11.0
-pycocotools>=2.0.7
-protobuf>=3.20.3,<4.0  # Specific range for compatibility
+# We'll install MatAnyone without dependencies so it doesn’t pull imageio==2.25.0.
+# See Dockerfile step below.
 
 # ============================================================================
 # API & ASYNC SUPPORT
@@ -125,11 +109,11 @@ aiohttp>=3.9.0,<4.0
 aiofiles>=23.2.0,<24.0
 httpx>=0.25.0,<1.0
 websockets>=12.0,<13.0
-redis>=5.0.0,<6.0  # For caching
-celery>=5.3.0,<6.0  # Task queue
+redis>=5.0.0,<6.0
+celery>=5.3.0,<6.0
 
 # ============================================================================
-# TESTING (Optional - move to requirements-dev.txt if preferred)
+# TESTING
 # ============================================================================
 pytest>=7.4.3,<8.0
 pytest-cov>=4.1.0,<5.0
@@ -137,10 +121,10 @@ pytest-asyncio>=0.21.1,<1.0
 pytest-benchmark>=4.0.0,<5.0
 
 # ============================================================================
-# CODE QUALITY (Optional - move to requirements-dev.txt if preferred)
+# CODE QUALITY
 # ============================================================================
 black>=23.11.0,<24.0
-ruff>=0.1.6,<1.0  # Fast Python linter
+ruff>=0.1.6,<1.0
 mypy>=1.7.0,<2.0
 pre-commit>=3.5.0,<4.0
 
@@ -150,15 +134,3 @@ pre-commit>=3.5.0,<4.0
 gunicorn>=21.2.0,<22.0
 supervisor>=4.2.5,<5.0
 python-crontab>=3.0.0,<4.0
-
-# ============================================================================
-# GIT DEPENDENCIES (Model Repositories)
-# ============================================================================
-# SAM2 - Latest stable
-git+https://github.com/facebookresearch/segment-anything-2.git
-
-# MatAnyone - Specific stable commit
-git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
-
-# Thin Plate Spline (if needed for MatAnyone)
-# git+https://github.com/cheind/py-thin-plate-spline