Update app.py

app.py

@@ -106,29 +106,6 @@ print('=' * 70)
 
 # =================================================================================================
 
-@spaces.GPU
-def inpaint_pitches(inp_seq,
-                    input_patch,
-                    input_inpaint_ratio,
-                    input_top_k
-                   ):
-    
-    print('*' * 70)
-    print('Inpainting pitches...')
-
-    m_pos = [i for i in range(0, SEQ_LEN) if (128*input_patch)+256 < melody_chords[i] < (128*(input_patch+1))+256]
-
-    results = predict_masked_tokens(model, inp_seq[:SEQ_LEN], mask_positions=m_pos)
-
-    output_seq = results['predicted_ids']
-
-    print('Done!')
-    print('=' * 70)
-    
-    return output_seq
-                       
-# =================================================================================================
-
 def load_midi(input_midi):
     
     """Process the input MIDI file and create a token sequence."""
@@ -296,6 +273,31 @@ def save_midi(tokens):
 
 # =================================================================================================
 
+@spaces.GPU
+def inpaint_pitches(inp_seq,
+                    input_patch,
+                    input_inpaint_ratio,
+                    input_top_k
+                   ):
+    
+    print('*' * 70)
+    print('Inpainting pitches...')
+
+    inp_seq = inp_seq[:SEQ_LEN]
+
+    m_pos = [i for i in range(0, SEQ_LEN) if (128*input_patch)+256 < inp_seq[i] < (128*(input_patch+1))+256]
+
+    results = predict_masked_tokens(model, inp_seq, mask_positions=m_pos)
+
+    output_seq = results['predicted_ids']
+
+    print('Done!')
+    print('=' * 70)
+    
+    return output_seq
+                       
+# =================================================================================================
+
 def Inpaint_Pitches(input_midi,
                     input_patch,
                     input_inpaint_ratio,
@@ -426,7 +428,7 @@ with app:
 
     input_top_k = gr.Slider(1, 5, value=1, step=1, label="Top k value")
     
-    run_btn = gr.Button("inpaint", variant="primary")
+    run_btn = gr.Button("Inpaint Pitches", variant="primary")
 
     gr.Markdown("## Generation results")