Update app.py

app.py

@@ -2,6 +2,8 @@ import re, os
 import requests
 import json
 
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+
 headers = {
     "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36"
 }
@@ -72,17 +74,18 @@ for name in os.listdir(weight_uvr5_root):
         uvr5_names.append(name.replace(".pth", ""))
 
 func = AudioPre
+
 pre_fun_hp2 = func(
   agg=int(10),
   model_path=os.path.join(weight_uvr5_root, "UVR-HP2.pth"),
-  device="cuda",
+  device=device,
   is_half=True,
 )
 
 pre_fun_hp5 = func(
   agg=int(10),
   model_path=os.path.join(weight_uvr5_root, "UVR-HP5.pth"),
-  device="cuda",
+  device=device,
   is_half=True,
 )