Update requirements.txt

requirements.txt

@@ -1,14 +1,13 @@
 # Core dependencies
 gradio>=4.0.0
 spaces
-torch>=2.1.0
-torchvision>=0.16.0
-torchaudio>=2.1.0
+torch==2.5.1
+torchvision==0.20.1
+torchaudio==2.5.1
 
 # Flash Attention (required for SteadyDancer)
-# Using pre-built wheel for CUDA 12.1 + PyTorch 2.5 + Python 3.10
---find-links https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1
-flash-attn
+# Pre-built wheel for CUDA 12 + PyTorch 2.5 + Python 3.10
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
 # Model dependencies
 transformers>=4.36.0