Upload app.py

app.py

@@ -1,12 +1,12 @@
 #==================================================================================
-# https://huggingface.co/spaces/asigalov61/Guided-Accompaniment-Transformer
+# https://huggingface.co/spaces/asigalov61/MIDI-Genre-Classifier
 #==================================================================================
 
 print('=' * 70)
-print('Guided Accompaniment Transformer Gradio App')
+print('MIDI Genre Classifier Gradio App')
 
 print('=' * 70)
-print('Loading core Guided Accompaniment Transformer modules...')
+print('Loading core MIDI Genre Classifier modules...')
 
 import os
 import copy
@@ -16,7 +16,7 @@ import datetime
 from pytz import timezone
 
 print('=' * 70)
-print('Loading main Guided Accompaniment Transformer modules...')
+print('Loading main MIDI Genre Classifier modules...')
 
 os.environ['USE_FLASH_ATTENTION'] = '1'
 
@@ -30,6 +30,8 @@ torch.backends.cuda.enable_math_sdp(True)
 torch.backends.cuda.enable_flash_sdp(True)
 torch.backends.cuda.enable_cudnn_sdp(True)
 
+import numpy as np
+
 from huggingface_hub import hf_hub_download
 
 import TMIDIX
@@ -43,7 +45,7 @@ import random
 import tqdm
 
 print('=' * 70)
-print('Loading aux Guided Accompaniment Transformer modules...')
+print('Loading aux MIDI Genre Classifier modules...')
 
 import matplotlib.pyplot as plt
 
@@ -59,7 +61,7 @@ print('=' * 70)
 
 #==================================================================================
 
-MODEL_CHECKPOINT = 'Guided_Accompaniment_Transformer_Trained_Model_36457_steps_0.5384_loss_0.8417_acc.pth'
+MODEL_CHECKPOINT = 'MIDI_Genre_Classifier_Trained_Model_36457_steps_0.5384_loss_0.8417_acc.pth'
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
@@ -72,7 +74,7 @@ MAX_GEN_TOKS = 3072
 print('=' * 70)
 print('Loading popular hook melodies dataset...')
 
-popular_hook_melodies_pickle = hf_hub_download(repo_id='asigalov61/Guided-Accompaniment-Transformer', 
+popular_hook_melodies_pickle = hf_hub_download(repo_id='asigalov61/MIDI-Genre-Classifier', 
                                                filename='popular_hook_melodies_24_64_CC_BY_NC_SA.pickle'
                                               )
 
@@ -112,7 +114,7 @@ model = AutoregressiveWrapper(model, ignore_index=PAD_IDX, pad_value=PAD_IDX)
 print('=' * 70)
 print('Loading model checkpoint...')      
 
-model_checkpoint = hf_hub_download(repo_id='asigalov61/Guided-Accompaniment-Transformer', filename=MODEL_CHECKPOINT)
+model_checkpoint = hf_hub_download(repo_id='asigalov61/MIDI-Genre-Classifier', filename=MODEL_CHECKPOINT)
 
 model.load_state_dict(torch.load(model_checkpoint, map_location='cpu', weights_only=True))
 
@@ -348,10 +350,10 @@ def Generate_Accompaniment(input_midi,
     
                 song_f.append(['note', time, dur, channel, pitch, vel, patch])
 
-    fn1 = "Guided-Accompaniment-Transformer-Composition"
+    fn1 = "MIDI-Genre-Classifier-Composition"
     
     detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
-                                                              output_signature = 'Guided Accompaniment Transformer',
+                                                              output_signature = 'MIDI Genre Classifier',
                                                               output_file_name = fn1,
                                                               track_name='Project Los Angeles',
                                                               list_of_MIDI_patches=patches_map
@@ -405,11 +407,11 @@ with gr.Blocks() as demo:
 
     #==================================================================================
 
-    gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Guided Accompaniment Transformer</h1>")
+    gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>MIDI Genre Classifier</h1>")
     gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Guided melody accompaniment generation with transformers</h1>")
     gr.HTML("""            
             <p> 
-                <a href="https://huggingface.co/spaces/asigalov61/Guided-Accompaniment-Transformer?duplicate=true">
+                <a href="https://huggingface.co/spaces/asigalov61/MIDI-Genre-Classifier?duplicate=true">
                     <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-md.svg" alt="Duplicate in Hugging Face">
                 </a>
             </p>