Update app.py

app.py

@@ -1,4 +1,6 @@
 #==================================================================================
+# https://huggingface.co/spaces/asigalov61/Giant-Music-Transformer
+#==================================================================================
 
 print('=' * 70)
 print('Loading core Giant Music Transformer modules...')
@@ -21,9 +23,6 @@ torch.backends.cuda.enable_math_sdp(True)
 torch.backends.cuda.enable_flash_sdp(True)
 torch.backends.cuda.enable_cudnn_sdp(True)
 
-os.chdir('/home/ubuntu/Giant-Music-Transformer/')
-print("Current working directory: ", os.getcwd())
-sys.path.append(os.getcwd())
 import TMIDIX
 
 from midi_to_colab_audio import midi_to_colab_audio
@@ -32,7 +31,6 @@ from x_transformer_1_23_2 import *
 
 import random
 
-os.chdir('/home/ubuntu/')
 print('=' * 70)
 print('Loading aux Giant Music Transformer modules...')
 
@@ -77,7 +75,7 @@ model = AutoregressiveWrapper(model, ignore_index=PAD_IDX, pad_value=PAD_IDX)
 print('=' * 70)
 print('Loading model checkpoint...')
 
-model_path = '/home/ubuntu/Giant-Music-Transformer/Models/Medium/Giant_Music_Transformer_Medium_Trained_Model_10446_steps_0.7202_loss_0.8233_acc.pth'
+model_path = 'Giant-Music-Transformer/Models/Medium/Giant_Music_Transformer_Medium_Trained_Model_10446_steps_0.7202_loss_0.8233_acc.pth'
 
 model.load_state_dict(torch.load(model_path))
 
@@ -93,7 +91,7 @@ print('=' * 70)
 
 #==================================================================================
 
-SOUDFONT_PATH = '/usr/share/sounds/sf2/FluidR3_GM.sf2'
+SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7'
 
 NUM_OUT_BATCHES = 8
 
@@ -234,10 +232,10 @@ def save_midi(tokens, batch_number=None):
     patches = [0 if x==-1 else x for x in patches]
 
     if batch_number == None:
-        fname = '/home/ubuntu/Giant-Music-Transformer-Music-Composition'
+        fname = 'Giant-Music-Transformer-Music-Composition'
         
     else:
-        fname = '/home/ubuntu/Giant-Music-Transformer-Music-Composition_'+str(batch_number)
+        fname = 'Giant-Music-Transformer-Music-Composition_'+str(batch_number)
     
     data = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
                                                   output_signature = 'Giant Music Transformer',
@@ -338,7 +336,7 @@ def generate_callback(input_midi,
         midi_plot = TMIDIX.plot_ms_SONG(midi_score, plot_title='Batch # ' + str(i), return_plt=True)
 
         # File name
-        fname = '/home/ubuntu/Giant-Music-Transformer-Music-Composition_'+str(i)
+        fname = 'Giant-Music-Transformer-Music-Composition_'+str(i)
         
         # Save audio to a temporary file
         midi_audio = midi_to_colab_audio(fname + '.mid',