Update app.py

app.py

@@ -1,13 +1,9 @@
 import re, os
 import requests
 import json
+import torch
 
-import urllib.request
-urllib.request.urlretrieve("https://download.openxlab.org.cn/repos/file/Kevin676/NeuCoSVC-2/main?filepath=WavLM-Large.pt&sign=971a42d686a15fcd3aafae29c1c97220&nonce=1715413418821", "ckpt/WavLM-Large.pt")
-urllib.request.urlretrieve("https://download.openxlab.org.cn/repos/file/Kevin676/NeuCoSVC-2/main?filepath=G_150k.pt&sign=07d507c87b7b9ed63556b7c094e6b0b5&nonce=1715413397933", "ckpt/G_150k.pt")
-urllib.request.urlretrieve("https://download.openxlab.org.cn/repos/file/Kevin676/NeuCoSVC-v2/main?filepath=speech_XXL_cond.zip&sign=0520b3273355818d1ebee030bce88ee4&nonce=1715413443250", "speech_XXL_cond.zip")
-urllib.request.urlretrieve("https://download.openxlab.org.cn/models/Kevin676/rvc-models/weight/UVR-HP2.pth", "uvr5/uvr_model/UVR-HP2.pth")
-urllib.request.urlretrieve("https://download.openxlab.org.cn/models/Kevin676/rvc-models/weight/UVR-HP5.pth", "uvr5/uvr_model/UVR-HP5.pth")
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 import zipfile
 with zipfile.ZipFile("speech_XXL_cond.zip", 'r') as zip_ref:
@@ -86,14 +82,14 @@ 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,
 )