Update app.py

app.py

@@ -485,10 +485,16 @@ print('=' * 70)
 print('App start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
 print('=' * 70)
 
+#==================================================================================
+
 with gr.Blocks() as demo:
 
+    #==================================================================================
+
     demo.load(reset_demo)
 
+    #==================================================================================
+
     gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Monster Piano Transformer</h1>")
     gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Ultra-fast and very well fitted solo Piano music transformer</h1>")
     gr.HTML("""
@@ -496,7 +502,7 @@ with gr.Blocks() as demo:
             
             <p>
                 <a href="https://pypi.org/project/monsterpianotransformer/">
-                    <img src="https://upload.wikimedia.org/wikipedia/commons/6/64/PyPI_logo.svg" alt="PyPI Project">
+                    <img src="https://upload.wikimedia.org/wikipedia/commons/6/64/PyPI_logo.svg" alt="PyPI Project" style="width: 100px; height: auto;">
                 </a> or 
                 <a href="https://huggingface.co/spaces/asigalov61/Monster-Piano-Transformer?duplicate=true">
                     <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-md.svg" alt="Duplicate in Hugging Face">
@@ -505,8 +511,7 @@ with gr.Blocks() as demo:
             
             for faster execution and endless generation!
             """)
-
-
+    
     #==================================================================================
 
     final_composition = gr.State([])
@@ -580,12 +585,18 @@ with gr.Blocks() as demo:
     final_plot_output = gr.Plot(label="Final MIDI plot")
     final_file_output = gr.File(label="Final MIDI file")
 
+    #==================================================================================
+
     add_btn.click(add_batch, [batch_number, final_composition, generated_batches, block_lines, model_state],
-                  [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines])       
+                  [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines])
+
+    #==================================================================================
 
     remove_btn.click(remove_batch, [batch_number, num_gen_tokens, final_composition, generated_batches, block_lines, model_state], 
                      [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines])
 
+    #==================================================================================
+
     demo.unload(reset_demo)
 
 #==================================================================================