Update requirements.txt

requirements.txt

@@ -1,32 +1,26 @@
 --extra-index-url https://download.pytorch.org/whl/cu118 
 
-# Core Dependencies
-gradio==5.33.2
+# First install PyTorch from CUDA index
 torch==2.1.0+cu118
 torchvision==0.16.0+cu118
+
+# Then install everything else
+gradio==5.33.2
 transformers==4.44.0
 accelerate==0.24.1
 pillow==10.0.1
 requests==2.31.0
-numpy==1.26.4  # Avoids conflict with TensorFlow and older packages
-
-# NLP / Embedding Models
+numpy==1.26.4
 sentence-transformers==2.2.2
 einops==0.8.0
 timm==1.0.3
-
-# Computer Vision
 opencv-python==4.8.1.78
 opencv-contrib-python==4.8.1.78
 mediapipe==0.10.7
 retina-face==0.0.13
 gdown==4.7.1
 tqdm==4.66.1
-
-# DeepFace & TensorFlow Support
 deepface==0.0.79
 tf-keras==2.15.0
-tensorflow==2.17.0  # Pin to compatible version with deepface and CUDA 11.8
-
-# Flash Attention (use prebuilt wheel)
+tensorflow==2.17.0
 flash-attn==2.7.4.post1 --no-build-isolation