Update requirements.txt

requirements.txt

@@ -1,151 +1,37 @@
-# ============================================================================
-# PYTORCH CUDA WHEELS (CUDA 12.1) - Always keep at the top for Hugging Face
-# ============================================================================
+# CUDA support for T4
 --extra-index-url https://download.pytorch.org/whl/cu121
 
-# ============================================================================
-# CORE PYTHON DEPENDENCIES (Python 3.10)
-# ============================================================================
+# Core
 numpy==1.26.4
 Pillow>=10.0.1,<11.0
-setuptools>=65.7.0,<69.0
-wheel>=0.40.0,<1.0
-typing-extensions>=4.12.2,<5.0
 
-# ============================================================================
-# WEB FRAMEWORK & UI
-# ============================================================================
+# UI
 gradio==4.44.0
-gradio_client==1.3.0
-fastapi>=0.104.0,<1.0
-uvicorn[standard]>=0.24.0,<1.0
-python-multipart>=0.0.6,<1.0
 
-# ============================================================================
-# DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1)
-# ============================================================================
+# PyTorch for T4 GPU
 torch==2.5.1
 torchvision==0.20.1
-torchaudio==2.5.1
+
+# Models
 transformers>=4.45.0,<5.0
 huggingface_hub>=0.25.0,<1.0
-# Bump Accelerate so Diffusers offloading & memory features are available
 accelerate>=0.30.0,<1.0
-safetensors==0.4.3
-tokenizers>=0.19.0,<1.0
-datasets>=2.14.0,<3.0
 einops==0.8.0
-timm>=0.9.16,<1.0
-# xFormers optional; if no wheel for your env, Diffusers will still run without it
-xformers>=0.0.28,<0.1.0
-
-# >>> NEW: Diffusers for AI background generation <
-diffusers>=0.30.0,<0.31
 
-# ============================================================================
-# COMPUTER VISION & VIDEO PROCESSING
-# ============================================================================
-opencv-contrib-python-headless==4.10.0.84
-git+https://github.com/facebookresearch/segment-anything-2.git
-# Fix for imageio version pinning to avoid conflicts with moviepy & scikit-image:
+# Video processing
+opencv-python-headless==4.8.1.78  # CHANGED: Safer version without WASM
 imageio==2.25.0
 imageio-ffmpeg>=0.4.9,<1.0
 ffmpeg-python>=0.2.0,<1.0
-av>=11.0.0,<12.0
 moviepy>=1.0.3,<2.0
-albumentations>=1.3.1,<2.0
-
-# ============================================================================
-# SCIENTIFIC COMPUTING
-# ============================================================================
-scipy==1.13.1
-pandas>=2.0.3,<3.0
-matplotlib>=3.7.2,<4.0
-seaborn>=0.12.2,<1.0
-tqdm>=4.66.1,<5.0
-
-# ============================================================================
-# CONFIGURATION & UTILITIES
-# ============================================================================
-hydra-core==1.3.2
-omegaconf==2.3.0
-python-dotenv>=1.0.0,<2.0
-pyyaml>=6.0.1,<7.0
-click>=8.1.7,<9.0
-rich>=13.5.0,<14.0
-typer>=0.9.0,<1.0
-
-# ============================================================================
-# PERFORMANCE & OPTIMIZATION
-# ============================================================================
-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
-pynvml>=11.5.0,<12.0
-onnx>=1.14.0,<2.0
-onnxruntime>=1.16.0,<2.0
 
-# ============================================================================
-# LOGGING & MONITORING
-# ============================================================================
-loguru>=0.7.2,<1.0
-tensorboard>=2.15.0,<3.0
-wandb>=0.16.0,<1.0
-prometheus-client>=0.19.0,<1.0
-
-# ============================================================================
-# ERROR HANDLING & VALIDATION
-# ============================================================================
-pydantic>=2.5.0,<3.0
-marshmallow>=3.20.0,<4.0
-cerberus>=1.3.5,<2.0
-
-# ============================================================================
-# MATANYONE - UPDATED TO USE MAIN BRANCH (Fixes tensor dimension issue)
-# ============================================================================
+# SAM2 and MatAnyone
+git+https://github.com/facebookresearch/segment-anything-2.git
 git+https://github.com/pq-yang/MatAnyone.git@main
 
-# ============================================================================
-# API & ASYNC SUPPORT
-# ============================================================================
-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
-celery>=5.3.0,<6.0
-
-# ============================================================================
-# TESTING
-# ============================================================================
-pytest>=7.4.3,<8.0
-pytest-cov>=4.1.0,<5.0
-pytest-asyncio>=0.21.1,<1.0
-pytest-benchmark>=4.0.0,<5.0
-
-# ============================================================================
-# CODE QUALITY
-# ============================================================================
-black>=23.11.0,<24.0
-ruff>=0.1.6,<1.0
-mypy>=1.7.0,<2.0
-pre-commit>=3.5.0,<4.0
-
-# ============================================================================
-# DEPLOYMENT & PRODUCTION
-# ============================================================================
-gunicorn>=21.2.0,<22.0
-supervisor>=4.2.5,<5.0
-python-crontab>=3.0.0,<4.0
+# Scientific computing
+scipy==1.13.1
+tqdm>=4.66.1,<5.0
 
-# ============================================================================
-# NOTES / FIXES
-# ============================================================================
-# 1. Keep --extra-index-url at the top.
-# 2. Install MatAnyone from GitHub main branch (fixes tensor dimension issue).
-# 3. Imageio pinned to 2.25.0 to avoid moviepy/scikit-image breakage.
-# 4. If CUDA issues: clear HF cache and restart Space.
-# 5. Diffusers added for AI background creation; Accelerate bumped to enable memory offload features.
-# 6. Gradio pinned to 4.44.0 with compatible client for stability.
+# Optional for AI backgrounds
+diffusers>=0.30.0,<0.31