Upload app.py

app.py

@@ -6,7 +6,7 @@ print('=' * 70)
 print('Guided Accompaniment Transformer Gradio App')
 
 print('=' * 70)
-print('Loading core Monster Piano Transformer modules...')
+print('Loading core Guided Accompaniment Transformer modules...')
 
 import os
 
@@ -15,7 +15,7 @@ import datetime
 from pytz import timezone
 
 print('=' * 70)
-print('Loading main Monster Piano Transformer modules...')
+print('Loading main Guided Accompaniment Transformer modules...')
 
 os.environ['USE_FLASH_ATTENTION'] = '1'
 
@@ -40,7 +40,7 @@ from x_transformer_1_23_2 import *
 import random
 
 print('=' * 70)
-print('Loading aux Monster Piano Transformer modules...')
+print('Loading aux Guided Accompaniment Transformer modules...')
 
 import matplotlib.pyplot as plt
 
@@ -105,7 +105,7 @@ def load_model(model_selector):
     print('=' * 70)
     print('Loading model checkpoint...')      
     
-    model_checkpoint = hf_hub_download(repo_id='asigalov61/Monster-Piano-Transformer', filename=MODEL_CHECKPOINTS[model_selector])
+    model_checkpoint = hf_hub_download(repo_id='asigalov61/Guided-Accompaniment-Transformer', filename=MODEL_CHECKPOINTS[model_selector])
     
     model.load_state_dict(torch.load(model_checkpoint, map_location='cpu', weights_only=True))
     
@@ -188,13 +188,13 @@ def save_midi(tokens, batch_number=None, model_selector=''):
                 song_f.append(['note', time, dur, 0, pitch, vel, 0])
 
     if batch_number == None:
-        fname = 'Monster-Piano-Transformer-Music-Composition'
+        fname = 'Guided-Accompaniment-Transformer-Music-Composition'
         
     else:
-        fname = 'Monster-Piano-Transformer-Music-Composition_'+str(batch_number)
+        fname = 'Guided-Accompaniment-Transformer-Music-Composition_'+str(batch_number)
     
     data = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
-                                                  output_signature = 'Monster Piano Transformer',
+                                                  output_signature = 'Guided Accompaniment Transformer',
                                                   output_file_name = fname,
                                                   track_name='Project Los Angeles',
                                                   list_of_MIDI_patches=patches,
@@ -307,7 +307,7 @@ def generate_callback(input_midi,
                                            )
 
         # File name
-        fname = 'Monster-Piano-Transformer-Music-Composition_'+str(i)
+        fname = 'Guided-Accompaniment-Transformer-Music-Composition_'+str(i)
         
         # Save audio to a temporary file
         midi_audio = midi_to_colab_audio(fname + '.mid', 
@@ -403,12 +403,12 @@ def add_batch(batch_number, final_composition, generated_batches, block_lines, m
 
         # MIDI plot
         midi_plot = TMIDIX.plot_ms_SONG(midi_score, 
-                                        plot_title='Monster Piano Transformer Composition',
+                                        plot_title='Guided Accompaniment Transformer Composition',
                                         block_lines_times_list=block_lines[:-1],
                                         return_plt=True)
         
         # File name
-        fname = 'Monster-Piano-Transformer-Music-Composition'
+        fname = 'Guided-Accompaniment-Transformer-Music-Composition'
         
         # Save audio to a temporary file
         midi_audio = midi_to_colab_audio(fname + '.mid', 
@@ -440,12 +440,12 @@ def remove_batch(batch_number, num_tokens, final_composition, generated_batches,
     
         # MIDI plot
         midi_plot = TMIDIX.plot_ms_SONG(midi_score, 
-                                        plot_title='Monster Piano Transformer Composition',
+                                        plot_title='Guided Accompaniment Transformer Composition',
                                         block_lines_times_list=block_lines[:-1],
                                         return_plt=True)
     
         # File name
-        fname = 'Monster-Piano-Transformer-Music-Composition'
+        fname = 'Guided-Accompaniment-Transformer-Music-Composition'
         
         # Save audio to a temporary file
         midi_audio = midi_to_colab_audio(fname + '.mid', 
@@ -500,16 +500,16 @@ with gr.Blocks() as 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.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Guided Accompaniment Transformer</h1>")
+    gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Guided melody accompaniment generation with transformers</h1>")
     gr.HTML("""
-            Check out <a href="https://github.com/asigalov61/monsterpianotransformer">Monster Piano Transformer</a> on GitHub or on
+            Check out <a href="https://github.com/asigalov61/monsterpianotransformer">Guided Accompaniment Transformer</a> on GitHub or on
             
             <p>
                 <a href="https://pypi.org/project/monsterpianotransformer/">
                     <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">
+                <a href="https://huggingface.co/spaces/asigalov61/Guided-Accompaniment-Transformer?duplicate=true">
                     <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-md.svg" alt="Duplicate in Hugging Face">
                 </a>
             </p>