Update app.py

app.py

@@ -1,4 +1,7 @@
+import gradio as gr
 import note_seq
+
+from bokeh.embed import file_html
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("TristanBehrens/js-fakes-4bars")
@@ -116,14 +119,16 @@ def process(text):
   note_sequence = token_sequence_to_note_sequence(generated_sequence)
   synth = note_seq.midi_synth.synthesize
   array_of_floats = synth(note_sequence, sample_rate=SAMPLE_RATE)
-  return SAMPLE_RATE, array_of_floats
+  note_plot = note_seq.plot_sequence(note_sequence)
+  html = file_html(note_plot)
+  return (SAMPLE_RATE, array_of_floats), html
   
 title = "Music generation with GPT-2"
 
 iface = gr.Interface(
   fn=process, 
   inputs=[gr.inputs.Textbox(placeholder="PIECE_START")],
-  outputs='audio',
+  outputs=['audio', 'html'],
   title=title,
   examples=[["PIECE_START"], ["PIECE_START STYLE=JSFAKES GENRE=JSFAKES TRACK_START INST=48 BAR_START NOTE_ON=61"]]
 )