Update requirements.txt

requirements.txt

@@ -1,13 +1,11 @@
 # ============================================
-# PRODUCTION REQUIREMENTS - FINAL VERSION
-# MatAnyone (BEST) + SAM2 (VIDEO) + Rembg (FAST)
-# Optimized for HuggingFace Spaces L4 GPU
+# FIXED REQUIREMENTS - MatAnyone compatible
 # ============================================
 
 # Core Dependencies
 streamlit==1.32.2
-opencv-python-headless==4.8.1.78  # Fixed for compatibility
-opencv-contrib-python==4.8.1.78   # Matching version
+opencv-python-headless==4.8.1.78
+opencv-contrib-python==4.8.1.78
 numpy==1.26.4
 Pillow==10.2.0
 
@@ -17,113 +15,96 @@ GPUtil==1.4.0
 nvidia-ml-py==12.535.161
 py3nvml==0.2.7
 
-# PyTorch with CUDA (Latest stable)
-torch==2.2.0
-torchvision==0.17.0
-torchaudio==2.2.0
+# NOTE: PyTorch handled in Dockerfile
+# Don't include torch/torchvision/torchaudio here
 
 # ============================================
-# PRIMARY METHOD: MatAnyone (Production Quality)
+# PRIMARY: MatAnyone (with its dependencies)
 # ============================================
 git+https://github.com/pq-yang/MatAnyone.git
 
 # ============================================
-# SECONDARY METHOD: SAM2 (Video Optimized)
+# SECONDARY: SAM2
 # ============================================
 git+https://github.com/facebookresearch/segment-anything-2.git
 
 # ============================================
-# TERTIARY METHOD: Rembg (Fast Fallback)
+# TERTIARY: Rembg
 # ============================================
 rembg==2.0.53
 onnxruntime-gpu==1.16.3
 
 # ============================================
-# Supporting Libraries
+# Supporting Libraries - NO VERSION PINNING!
+# Let pip resolve versions automatically
 # ============================================
 
-# Transformers & HuggingFace (for model downloads)
-transformers==4.37.2
-accelerate==0.26.1
-huggingface-hub==0.20.3
-safetensors==0.4.2
-tokenizers==0.15.1
+# Transformers & HuggingFace
+transformers
+accelerate
+huggingface-hub
+safetensors
+tokenizers
 
 # Computer Vision
-timm==0.9.12
-albumentations==1.3.1
-ultralytics==8.1.18
-supervision==0.17.1
-
-# Image Processing
-scipy==1.12.0
-scikit-image==0.22.0
-matplotlib==3.8.2
-imageio==2.33.1
-imageio-ffmpeg==0.4.9
+timm
+albumentations
+ultralytics
+supervision
+
+# Image Processing - NO VERSION for imageio!
+scipy
+scikit-image
+matplotlib
+imageio  # NO VERSION - let MatAnyone decide!
+imageio-ffmpeg
 
 # Video Processing
-moviepy==1.0.3
-av==11.0.0
-decord==0.6.0
+moviepy
+av
 
 # ML Libraries
-xgboost==2.0.3
-lightgbm==4.3.0
-scikit-learn==1.4.0
+xgboost
+lightgbm
+scikit-learn
 
 # Data Processing
-pandas==2.2.0
-pyarrow==15.0.0
+pandas
+pyarrow
 
 # Progress & UI
-tqdm==4.66.1
-rich==13.7.0
+tqdm
+rich
 
 # Web & Downloads
-requests==2.31.0
-httpx==0.26.0
-aiohttp==3.9.3
+requests
+httpx
+aiohttp
 
 # Memory Management
-memory-profiler==0.61.0
-pympler==1.0.1
-objgraph==3.6.1
+memory-profiler
+pympler
+objgraph
 
 # Parallel Processing
-joblib==1.3.2
-threadpoolctl==3.2.0
+joblib
+threadpoolctl
 
 # Utilities
-python-dotenv==1.0.1
-fire==0.5.0
-typer==0.9.0
+python-dotenv
+fire
+typer
 
-# ============================================
-# OPTIONAL: Additional Features
-# ============================================
-
-# Audio support (for videos with sound)
-librosa==0.10.1
-soundfile==0.12.1
-pydub==0.25.1
+# Audio support
+librosa
+soundfile
+pydub
 
 # Advanced CV models
-open-clip-torch==2.24.0
-clip-interrogator==0.6.0
-groundingdino-py==0.4.0
+open-clip-torch
+clip-interrogator
+groundingdino-py
 
 # Monitoring
-wandb==0.16.3
-tensorboard==2.15.1
-
-# ============================================
-# NOTES:
-# 1. MatAnyone provides best quality with temporal consistency
-# 2. SAM2 is video-optimized segmentation
-# 3. Rembg is fast fallback
-# 4. OpenCV is always available (comes with opencv-python)
-# 5. Models download dynamically on first use
-# 6. Total package size: ~10-15GB
-# 7. Models add: 500MB-2GB depending on selection
-# ============================================
+wandb
+tensorboard