Update app.py

app.py

@@ -380,7 +380,7 @@ def demucs_separator(audio, model_key, seg_size, shifts, overlap, segments_enabl
 
         stems = [os.path.join(out_dir, file_name) for file_name in separation]
 
-        if model == "htdemucs_6s.yaml":
+        if model_key == "htdemucs_6s":
             return stems[0], stems[1], stems[2], stems[3], stems[4], stems[5]
         else:
             return stems[0], stems[1], stems[2], stems[3], None, None
@@ -389,7 +389,7 @@ def demucs_separator(audio, model_key, seg_size, shifts, overlap, segments_enabl
 
 def update_stems(model):
     """Update the visibility of stem outputs based on the selected Demucs model."""
-    if model == "htdemucs_6s.yaml":
+    if model == "htdemucs_6s":
         return gr.update(visible=True)
     else:
         return gr.update(visible=False)