Piano Guitar

app.py

@@ -12,7 +12,7 @@ def inference(audio):
     os.makedirs("out", exist_ok=True)
     write('test.wav', audio[0], audio[1])
 
-    command = "python3 -m demucs.separate -n mdx_extra_q -d cpu test.wav -o out"
+    command = "python3 -m demucs.separate -n htdemucs_6s -d cpu test.wav -o out"
     process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     print("Demucs script output:", process.stdout.decode())
   
@@ -35,6 +35,8 @@ def inference(audio):
              "./out/mdx_extra_q/test/bass.wav",
              "./out/mdx_extra_q/test/drums.wav",
              "./out/mdx_extra_q/test/other.wav",
+              "./out/mdx_extra_q/test/piano.wav",
+              "./out/mdx_extra_q/test/guitar.wav",
             ]
 
     
@@ -57,6 +59,8 @@ gr.Interface(
      gr.components.Audio(type="filepath", label="Bass"),
      gr.components.Audio(type="filepath", label="Drums"),
      gr.components.Audio(type="filepath", label="Other"),
+     gr.components.Audio(type="filepath", label="Piano"),
+     gr.components.Audio(type="filepath", label="Guitar"),
     ],
     title=title,
     description=description,