Update requirements.txt

requirements.txt

@@ -8,11 +8,12 @@
 --extra-index-url https://download.pytorch.org/whl/cu121
 
 # --- PyTorch (Must be first) ---
-torch==2.3.1
-torchvision==0.18.1
+torch==2.5.1
+torchvision==0.20.1
 
-# --- Core ---
+# --- Core deps (before models) ---
 numpy>=1.24.4
+einops==0.6.0
 Pillow==10.0.0
 protobuf==3.20.3
 
@@ -27,7 +28,6 @@ scikit-image==0.19.3
 # --- SAM2 Dependencies ---
 hydra-core==1.3.2
 omegaconf==2.3.0
-einops==0.6.0
 timm==0.9.0
 pyyaml==6.0.0
 matplotlib==3.5.0
@@ -36,10 +36,13 @@ iopath==0.1.10
 # --- SAM2 Installation ---
 git+https://github.com/facebookresearch/segment-anything-2.git
 
-# --- MatAnyone Dependencies ---
+# --- MatAnyone Dependencies (BEFORE MatAnyone) ---
+huggingface-hub==0.20.0
 kornia==0.7.2
 tqdm==4.66.1
 scipy==1.10.1
+transformers==4.30.0
+accelerate==0.20.0
 
 # --- MatAnyone Installation ---
 git+https://github.com/pq-yang/MatAnyone.git
@@ -47,7 +50,7 @@ git+https://github.com/pq-yang/MatAnyone.git
 # --- Streamlit UI ---
 streamlit==1.32.0
 
-# --- Web stack (for potential API extensions) ---
+# --- Web stack ---
 fastapi==0.104.0
 uvicorn==0.24.0
 
@@ -57,19 +60,12 @@ pydantic-core==2.16.1
 annotated-types==0.6.0
 typing-extensions==4.8.0
 
-# --- Helpers / Utilities ---
-huggingface-hub==0.20.0
+# --- Utilities ---
 ffmpeg-python==0.2.0
 psutil==5.9.4
 requests==2.31.0
 scikit-learn==1.3.0
 loguru==0.6.0
 python-multipart==0.0.5
-
-# --- Additional for SAM2 + MatAnyone ---
-transformers==4.30.0
-accelerate==0.20.0
 ninja==1.10.0
-
-# --- For better file handling ---
 aiofiles==23.1.0