Update the PyTorch version to CUDA 12.8 in the requirements.

README.md

@@ -4,7 +4,7 @@ emoji: 📈
 colorFrom: blue
 colorTo: gray
 sdk: gradio
-sdk_version: 5.39.0
+sdk_version: 5.44.1
 app_file: app.py
 pinned: true
 license: apache-2.0

requirements.txt

@@ -1,6 +1,6 @@
---extra-index-url https://download.pytorch.org/whl/cu124
+--extra-index-url https://download.pytorch.org/whl/cu128
 
-gradio==5.39.0
+gradio
 
 basicsr @ git+https://github.com/avan06/BasicSR
 facexlib @ git+https://github.com/avan06/facexlib
@@ -9,10 +9,8 @@ gfpgan @ git+https://github.com/avan06/GFPGAN
 numpy
 opencv-python
 
-torch==2.5.0+cu124; sys_platform != 'darwin'
-torchvision==0.20.0+cu124; sys_platform != 'darwin'
-torch==2.5.0; sys_platform == 'darwin'
-torchvision==0.20.0; sys_platform == 'darwin'
+torch
+torchvision
 
 scipy
 tqdm