Update requirements.txt

requirements.txt

@@ -1,42 +1,41 @@
-# Main dependencies
-torch>=2.6.0
-torchvision>=0.21.0
-lightning>=2.2.4
-numpy>=1.26.4
-matplotlib>=3.8.0
-pyyaml>=6.0.0
-
-# Huggingface
-huggingface-hub>=0.30.2
-
-# Image processing and computer vision
-kornia>=0.7.2
-opencv-python>=4.10.0.84
-opencv-contrib-python>=4.10.0.84
-einops>=0.8.0
-
-# Custom cuda implementation /modules/cupy_module/
-cupy-cuda12x>=12.0.0  # For CUDA 12.4
-# Note: For cupy, you need to install the specific version for your CUDA version
-# Examples:
-# cupy-cuda11x for CUDA 11.x
-# cupy-cuda12x for CUDA 12.x
-# cupy-cuda10x for CUDA 10.x
-
-# Utilities and tools
-scipy>=1.7.0
-tensorboard>=2.8.0
-
-# Project-Specific Dependencies
-# RAFT (Flow Estimation)
-# Note: RAFT is included in the project code; no external installation is required.
-
-# FLOLPIPS (Quality Metrics)
-# Note: FLOLPIPS is included in the project code; no external installation is required.
-
-# Gradio
-gradio>=4.34.0
-imageio>=2.34.1
-imageio-ffmpeg>=0.6.0
-
-
+# Main dependencies
+torch>=2.6.0
+torchvision>=0.21.0
+lightning>=2.2.4
+numpy>=1.26.4
+matplotlib>=3.8.0
+pyyaml>=6.0.0
+
+# Huggingface
+huggingface-hub>=0.30.2
+safetensors
+
+# Image processing and computer vision
+kornia>=0.7.2
+opencv-python>=4.10.0.84
+opencv-contrib-python>=4.10.0.84
+einops>=0.8.0
+
+# Custom cuda implementation /modules/cupy_module/
+cupy-cuda12x>=12.0.0  # For CUDA 12.4
+# Note: For cupy, you need to install the specific version for your CUDA version
+# Examples:
+# cupy-cuda11x for CUDA 11.x
+# cupy-cuda12x for CUDA 12.x
+# cupy-cuda10x for CUDA 10.x
+
+# Utilities and tools
+scipy>=1.7.0
+tensorboard>=2.8.0
+
+# Project-Specific Dependencies
+# RAFT (Flow Estimation)
+# Note: RAFT is included in the project code; no external installation is required.
+
+# FLOLPIPS (Quality Metrics)
+# Note: FLOLPIPS is included in the project code; no external installation is required.
+
+# Gradio
+gradio>=4.34.0
+imageio>=2.34.1
+imageio-ffmpeg>=0.6.0