Update app.py

app.py

@@ -13,6 +13,9 @@ from x_transformer_1_23_2 import *
 import random
 import tqdm
 
+import pprint
+import io
+
 from midi_to_colab_audio import midi_to_colab_audio
 import TMIDIX
 
@@ -67,7 +70,7 @@ def GenerateMusic():
     print('Done!')
     print('=' * 70)
 
-    input_num_tokens = 2040
+    input_num_tokens = 1024+512
     print('-' * 70)
 
     #===============================================================================
@@ -80,6 +83,11 @@ def GenerateMusic():
     
     def tokens2txt(tokens):
         return [chr(tok-648) for tok in tokens if 0+648 < tok < 128+648 ]
+
+    def pprint_to_string(obj, compact=True):
+        output = io.StringIO()
+        pprint.pprint(obj, stream=output, compact=compact)
+        return output.getvalue()
     
     print('=' * 70)
     print('Generating...')
@@ -126,7 +134,7 @@ def GenerateMusic():
     print('Sample INTs', out1[:12])
     print('=' * 70)
 
-    generated_song_description = '\n'.join(''.join(tokens2txt(out1)).split('. '))
+    generated_song_description = pprint_to_string(''.join(tokens2txt(out1)))
 
     if len(out1) != 0: