Update app.py

app.py

@@ -296,7 +296,7 @@ def inpaint_pitches(inp_seq,
     if input_inpaint_ratio < 1:
         m_pos = sorted(random.sample(m_pos, k=int(round(len(m_pos) * input_inpaint_ratio))))
 
-    results = predict_masked_tokens(model, inp_seq, mask_positions=m_pos, topk=1)
+    results = predict_masked_tokens_iter(model, inp_seq, mask_positions=m_pos, topk=1)
 
     output_seq = results['predicted_ids']