Update requirements.txt

requirements.txt

@@ -1,7 +1,7 @@
 --extra-index-url https://download.pytorch.org/whl/cu121 
 gradio==5.33.1
 torch==2.1.0
-torchvision==0.16.0
+torchvision==0.16.0+cu121
 transformers==4.44.0
 accelerate==0.24.1
 pillow==10.0.1
@@ -20,5 +20,5 @@ retina-face==0.0.13
 gdown==4.7.1
 tqdm==4.66.1
 
-# Flash Attention (use prebuilt wheel)
-flash-attn==2.7.4.post1
+# Flash Attention - use prebuilt wheel if available
+flash-attn==2.7.4.post1 --no-build-isolation