Update app.py

app.py

@@ -108,8 +108,9 @@ def rename_stems(input_file, output_dir, stems, output_format):
         renamed_stems.append(new_path)
     return renamed_stems
 
-def roformer_separator(audio, model, seg_size, overlap, model_dir, out_dir, out_format, norm_thresh, amp_thresh):
+def roformer_separator(audio, model_key, seg_size, overlap, model_dir, out_dir, out_format, norm_thresh, amp_thresh):
     """Separate audio using Roformer model."""
+    model = ROFORMER_MODELS[model_key]
     with tempfile.TemporaryDirectory() as tmp_dir:
         separator = Separator(
             model_file_dir=model_dir,