Update app.py

app.py

@@ -25,14 +25,9 @@ import TPLOTS
 from midi_to_colab_audio import midi_to_colab_audio
 
 # =================================================================================================
-                       
+
 @spaces.GPU
-def Generate_POP_Medley(input_num_medley_comps):
-    
-    print('=' * 70)
-    print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
-    start_time = reqtime.time()
-    print('=' * 70)
+def generate_MIDI_images(num_images):
     
     print('Loading model...')
 
@@ -72,13 +67,16 @@ def Generate_POP_Medley(input_num_medley_comps):
     print('=' * 70)
     print('Done!')
     print('=' * 70)
-
-    print('Req number of medley compositions:', input_num_medley_comps)
+    print('Generating...')
+    print('=' * 70)
+    
+    images = trainer.sample(batch_size = input_num_medley_comps, return_pil_images = True)
 
     print('=' * 70)
-    print('Generating...')
+    print('Done!')
+    print('=' * 70)
 
-    images = trainer.sample(batch_size = input_num_medley_comps, return_pil_images = True)
+    print('Processing...')
 
     threshold = 128
     
@@ -90,13 +88,30 @@ def Generate_POP_Medley(input_num_medley_comps):
       imgs_array.append(farr)
 
     print('Done!')
+
+    return imgs_array
+    
+# =================================================================================================
+
+def Generate_POP_Medley(input_num_medley_comps):
+    
+    print('=' * 70)
+    print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
+    start_time = reqtime.time()
+    print('=' * 70)
+    
+    print('Req number of medley compositions:', input_num_medley_comps)
     print('=' * 70)
     
     #===============================================================================
 
+    imgs_array = generate_MIDI_images(input_num_medley_comps)
+
+    #===============================================================================
+    
+    print('=' * 70)
     print('Converting images to scores...')
     
-
     medley_compositions_escores = []
 
     for i in imgs_array:
@@ -193,6 +208,8 @@ if __name__ == "__main__":
                     "This is a demo for MIDI Images dataset\n\n"
                     "Please see [MIDI Images](https://huggingface.co/datasets/asigalov61/MIDI-Images) Hugging Face repo for more information\n\n"
                     )        
+
+        gr.Markdown("Choose medley settings")
         
         input_num_medley_comps = gr.Slider(1, 12, value=8, step=1, label="Number of medley compositions")