Update requirements.txt

requirements.txt

@@ -1,25 +1,33 @@
+# Core deep learning
+torch==2.2.2+cu121
+torchvision==0.17.2+cu121
+--extra-index-url https://download.pytorch.org/whl/cu121
+
 diffusers==0.32.2
 accelerate>=1.1.1
 transformers>=4.46.2
 numpy==1.26.0
-torch>=2.5.0
-torchvision>=0.20.0
-opencv-python>=4.10.0.84
+einops
+pydantic>=2.10.3
+protobuf
+sentencepiece>=0.2.0
+
+# Image / Video processing
+opencv-python-headless>=4.10.0.84
 imageio>=2.35.1
 imageio-ffmpeg>=0.5.1
-sentencepiece>=0.2.0
-SwissArmyTransformer>=0.4.12
-openai>=1.54.0
-pydantic>=2.10.3
-einops
 decord
-protobuf
 scipy
 scikit-image
-qwen_vl_utils
-gdown
-numpy==1.26.0
 moviepy==2.2.1
-spandrel>=0.4.0
 scikit-video>=1.1.11
-gradio
+
+# Specialized libraries
+SwissArmyTransformer>=0.4.12
+spandrel>=0.4.0
+gdown
+qwen_vl_utils
+openai>=1.54.0
+
+# UI
+gradio