Update app.py

app.py

@@ -376,7 +376,7 @@ def load_hubert():
     hubert_model.load_state_dict(state_dict['model'])
     
     # Move the model to the desired device
-    hubert_model = hubert_model.to("cpu")
+    hubert_model = hubert_model.to("cuda:0")
     
     # Set the model to half precision if required
     if config.is_half:
@@ -387,7 +387,7 @@ def load_hubert():
     # Set the model to evaluation mode
     hubert_model.eval()
 
-load_hubert()
+#load_hubert()
 
 def rvc_models(model_name):
   global vc, net_g, index_files, tgt_sr, version
@@ -467,7 +467,7 @@ def infer_gpu(hubert_model, net_g, audio, f0_up_key, index_file, tgt_sr, version
     )
     
 def rvc_infer_music(url, model_name, song_name, split_model, f0_up_key, vocal_volume, inst_volume):
-  #load_hubert()
+  load_hubert()
   #print(hubert_model)
   url = url.strip().replace(" ", "")
   model_name = model_name.strip().replace(" ", "")