Update requirements.txt

requirements.txt

@@ -1,29 +1,19 @@
---extra-index-url https://download.pytorch.org/whl/cu124
+--extra-index-url https://download.pytorch.org/whl/cu121
+
 numpy==1.26.4
-gradio==5.9.1
 opencv-python-headless==4.10.0.84
+torch==2.2.2+cu121
+torchvision==0.17.2+cu121
 onnx==1.16.1
+onnxruntime-gpu==1.20.1
 insightface==0.7.3
-albucore==0.0.16
-psutil==5.9.6
-torch==2.5.1+cu124; sys_platform != 'darwin'
-torch==2.5.1; sys_platform == 'darwin'
-torchvision==0.20.1+cu124; sys_platform != 'darwin'
-torchvision==0.20.1; sys_platform == 'darwin'
-onnxruntime==1.20.1; sys_platform == 'darwin' and platform_machine != 'arm64'
-onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
-onnxruntime-gpu==1.20.1; sys_platform != 'darwin'
-tensorflow==2.14.0
-tk==0.1.0
-customtkinter==5.2.0
-tkinterdnd2==0.3.0; sys_platform != 'darwin' and platform_machine != 'arm64'
-tkinterdnd2-universal==1.7.3; sys_platform == 'darwin' and platform_machine == 'arm64'
-pillow==10.0.0
-protobuf==4.23.4
 gfpgan==1.3.8
-opennsfw2
-keras
+gradio==5.9.1
+pillow==10.0.0
+albumentations==1.4.0
 tqdm==4.66.4
-ftfy
+psutil==5.9.6
 regex
+ftfy
 pyvirtualcam
+