GPU version

app.py

@@ -12,7 +12,7 @@ def inference(audio):
 
     try:
         # Using subprocess.run for better control
-        command = f"python3 -m demucs.separate -n htdemucs_6s -d cpu {audio_path} -o out"
+        command = f"python3 -m demucs.separate -n htdemucs_6s -d gpu {audio_path} -o out"
         process = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         print("Demucs script output:", process.stdout.decode())
     except subprocess.CalledProcessError as e:
@@ -21,7 +21,7 @@ def inference(audio):
 
     try:
         # Separating the stems using your custom separator
-        separator = Separator("./out/htdemucs_6s/test/vocals.wav", model_name='UVR_MDXNET_KARA_2', use_cuda=False, output_format='mp3')
+        separator = Separator("./out/htdemucs_6s/test/vocals.wav", model_name='UVR_MDXNET_KARA_2', use_cuda=True, output_format='mp3')
         primary_stem_path, secondary_stem_path = separator.separate()
     except Exception as e:
         print("Error in custom separation:", str(e))