Update app.py

app.py

@@ -191,6 +191,7 @@ def save_midi(tokens, batch_number=None):
     """Convert token sequence back to a MIDI score and write it using TMIDIX (without velocity).
        The output MIDI file name incorporates a date-time stamp.
     """
+
     time = 0
     dur = 1
     vel = 90
@@ -203,6 +204,8 @@ def save_midi(tokens, batch_number=None):
     channels = [0] * 16
     channels[9] = 1
 
+    song_f = []
+
     for ss in tokens:
 
         if 0 <= ss < 256:
@@ -244,13 +247,8 @@ def save_midi(tokens, batch_number=None):
     
     # Generate a time stamp using the PDT timezone.
     timestamp = datetime.datetime.now(PDT).strftime("%Y%m%d_%H%M%S")
-    
-    '''if batch_number is None:
-        fname = f"Orpheus-Music-Transformer-Music-Composition_{timestamp}"
-    else:
-        fname = f"Orpheus-Music-Transformer-Music-Composition_{timestamp}_Batch_{batch_number}"'''
 
-    fname = f"Orpheus-Music-Transformer-Composition"
+    fname = f"Orpheus-Music-Transformer-Composition_" + str(batch_number)
     
     TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
         song_f,
@@ -422,7 +420,7 @@ with gr.Blocks() as demo:
 
     gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>Orpheus Music Transformer</h1>")
     gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>SOTA 8k multi-instrumental music transformer trained on 2.31M+ high-quality MIDIs</h1>")
-    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>(This is WIP preview. Please check back for final release soon.)</h1>")
+    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>This is a WIP preview. Please check back for final release soon.</h1>")
     gr.HTML("""
         Check out <a href="https://huggingface.co/datasets/projectlosangeles/Godzilla-MIDI-Dataset">Godzilla MIDI Dataset</a> on Hugging Face
         <p>