Adjust for 2 stem compatibility

app.py

@@ -67,14 +67,12 @@ custom_theme = PurpleTheme()
 def run_demucs(audio):
 	os.makedirs("out", exist_ok=True)
 	write('test.wav', audio[0], audio[1])
-	result = os.system("python3 -m demucs.separate -n htdemucs_ft -d cpu test.wav -o out")
+	result = os.system("python3 -m demucs.separate -n htdemucs_ft --two-stems=vocals -d cpu test.wav -o out")
 	print(f"Demucs result: {result}")
 
 	# Check if files exist before returning
 	files = ["./out/htdemucs_ft/test/vocals.wav",
-		"./out/htdemucs_ft/test/bass.wav",
-		"./out/htdemucs_ft/test/drums.wav",
-		"./out/htdemucs_ft/test/other.wav"]
+		"./out/htdemucs_ft/test/no_vocals.wav"]
 
 	for file in files:
 		if not os.path.isfile(file):
@@ -92,9 +90,7 @@ Gradio.Interface(
     run_demucs,
     Gradio.Audio(type="numpy", label="Input"),
     [Gradio.Audio(type="filepath", label="Vocals", interactive=False),
-     Gradio.Audio(type="filepath", label="Bass", interactive=False),
-     Gradio.Audio(type="filepath", label="Drums", interactive=False),
-     Gradio.Audio(type="filepath", label="Other", interactive=False)],
+     Gradio.Audio(type="filepath", label="Instrumental", interactive=False)],
     title=title,
     description=description,
     article=article,