Update gradio_mix.py

gradio_mix.py

@@ -61,7 +61,7 @@ def _pick_device():
     return "cuda" if torch.cuda.is_available() else "cpu"
 
 device = _pick_device()
-ASR_DEVICE = "cpu"  # force whisperx/pyannote to CPU to avoid cuDNN issues
+ASR_DEVICE = "cuda"  # force whisperx/pyannote to CPU to avoid cuDNN issues
 whisper_model, align_model = None, None
 tts_edit_model = None
 
@@ -109,7 +109,7 @@ class UVR5:
             model_path=model_path,
             audio_path = model_dir,
             result_path = model_dir,
-            device = 'cpu',
+            device = _pick_device(),
             process_method = "MDX-Net",
             # Keep base_dir and model_dir the same so all UVR5 metadata
             # (model_data.json, model_name_mapper.json, etc.) are resolved
@@ -118,7 +118,7 @@ class UVR5:
             **configs
         )
 
-        uvr5_model = Inference(model_data, 'cpu')
+        uvr5_model = Inference(model_data, _pick_device())
         uvr5_model.load_model(model_path, 1)
         return uvr5_model