Update app.py

app.py

@@ -168,7 +168,7 @@ def api_vamp(data):
 
 OUT_DIR = Path("gradio-outputs")
 OUT_DIR.mkdir(exist_ok=True)
-def harp_vamp(input_audio_file, periodic_p, n_mask_codebooks):
+def harp_vamp(input_audio_file, periodic_p, n_mask_codebooks, model_choice):
     sig = at.AudioSignal(input_audio_file)
     sr, samples = sig.sample_rate, sig.samples[0][0].detach().cpu().numpy()
     # convert to int32
@@ -176,7 +176,7 @@ def harp_vamp(input_audio_file, periodic_p, n_mask_codebooks):
     sr, samples =  _vamp(
         seed=0,
         input_audio=(sr, samples),
-        model_choice=init_model_choice,
+        model_choice=model_choice,
         pitch_shift_amt=0,
         periodic_p=periodic_p,
         n_mask_codebooks=n_mask_codebooks,