Add write music21

app.py

@@ -74,9 +74,9 @@ def inference(yt_audio_path):
     piano_roll = create_image_from_note_sequence(est_ns)
     
     parsed = converter.parse("./transcribed.mid")
-    parsed.show()
-    
-    return "./transcribed.mid", (SAMPLE_RATE, int16_data), piano_roll
+    score = parsed.write()
+    print(score)
+    return "./transcribed.mid", (SAMPLE_RATE, int16_data), piano_roll, score
   
 title = "Transcribe music from YouTube videos using Transformers."
 description = """
@@ -125,12 +125,12 @@ with demo:
             midi_audio = gr.Audio()
         with gr.Row():
             piano_roll = gr.Image()
-        
+            score = gr.Image()
         btn.click(inference,
                   inputs=yt_audio_path,
-                  outputs=[midi_file, midi_audio, piano_roll])
+                  outputs=[midi_file, midi_audio, piano_roll, score])
         
-        gr.Markdown(article)
+    gr.Markdown(article)
         
 
 demo.launch()