Update app.py

app.py

@@ -167,7 +167,6 @@ def load_midi(input_midi):
         print('Score hss', len(melody_chords), 'tokens')
         print('=' * 70)
 
-
         return melody_chords
 
     else:
@@ -176,10 +175,16 @@ def load_midi(input_midi):
 #==================================================================================
 
 @spaces.GPU
-def Generate_Music_Bridge(input_midi,
-                          model_temperature,
-                          model_sampling_top_p
-                         ):
+def Inpaint_Pitches(patches_to_inpaint,
+                    inpaint_every_nth_note,
+                    forbid_original_pitches,
+                    max_inpainted_pitch_dev,
+                    max_inpaint_tries_per_note,
+                    num_prime_tokens,
+                    num_mem_tokens,
+                    model_temperature,
+                    model_sampling_top_k
+                   ):
 
     #===============================================================================
 
@@ -188,20 +193,33 @@ def Generate_Music_Bridge(input_midi,
     start_time = reqtime.time()
     print('=' * 70)
 
-    print('=' * 70)
-    print('Requested settings:')
-    print('=' * 70)
-    fn = os.path.basename(input_midi)
-    fn1 = fn.split('.')[0]
-    print('Input MIDI file name:', fn)
-    print('Model temperature:', model_temperature)
-    print('Model top p:', model_sampling_top_p)
-   
-    print('=' * 70)
+    if input_midi is not None:
 
-    #==================================================================
+        print('=' * 70)
+        print('Requested settings:')
+        print('=' * 70)
+        fn = os.path.basename(input_midi)
+        fn1 = fn.split('.')[0]
+        print('Input MIDI file name:', fn)
+        print('-' * 70)
+    
+        print('Patches to inpaint:', patches_to_inpaint)
+        print('Inpaint every nth note:', inpaint_every_nth_note)
+        print('Forbid original pitches:', forbid_original_pitches)
+        print('Max inpainted pitch dev:', max_inpainted_pitch_dev)
+        print('Max inpaint tries per note:', max_inpaint_tries_per_note)
+        print('-' * 70)
+        print('Number of prime tokens:', num_prime_tokens)
+        print('Number of memory tokens:', num_mem_tokens)
+        print('-' * 70)
+        print('Model temperature:', model_temperature)
+        print('Model top p:', model_sampling_top_k)
+       
+        print('=' * 70)
+    
+        #==================================================================
 
-    if input_midi is not None:
+    
 
         print('Loading MIDI...')
     
@@ -460,11 +478,17 @@ with gr.Blocks() as demo:
     output_plot = gr.Plot(label="MIDI score plot")
     output_midi = gr.File(label="MIDI file", file_types=[".mid"])
 
-    generate_btn.click(Generate_Music_Bridge, 
-                       [input_midi,
+    generate_btn.click(Inpaint_Pitches, 
+                       [patches_to_inpaint,
+                        inpaint_every_nth_note,
+                        forbid_original_pitches,
+                        max_inpainted_pitch_dev,
+                        max_inpaint_tries_per_note,
+                        num_prime_tokens,
+                        num_mem_tokens,
                         model_temperature,
-                        model_sampling_top_p
-                       ], 
+                        model_sampling_top_k
+                       ],
                        [output_audio,
                         output_plot,
                         output_midi                          
@@ -472,17 +496,23 @@ with gr.Blocks() as demo:
                       )
 
     gr.Examples(
-                [["Sharing The Night Together.kar", 0.9, 0.96]
+                [["Orpheus-Music-Transformer-MI-Seed-1.mid", 0.9, 0.96]
                 ],
-                [input_midi,
+                [patches_to_inpaint,
+                 inpaint_every_nth_note,
+                 forbid_original_pitches,
+                 max_inpainted_pitch_dev,
+                 max_inpaint_tries_per_note,
+                 num_prime_tokens,
+                 num_mem_tokens,
                  model_temperature,
-                 model_sampling_top_p
+                 model_sampling_top_k
                 ],
                 [output_audio,
                  output_plot,
                  output_midi
                 ],
-                Generate_Music_Bridge
+                Inpaint_Pitches
     )
     
 #==================================================================================