Update app.py

app.py

@@ -4,6 +4,7 @@ import note_seq
 from bokeh.embed import file_html
 from bokeh.resources import CDN
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import scipy.io.wavfile
 
 tokenizer = AutoTokenizer.from_pretrained("TristanBehrens/js-fakes-4bars")
 model = AutoModelForCausalLM.from_pretrained("TristanBehrens/js-fakes-4bars")
@@ -122,7 +123,9 @@ def process(text):
   array_of_floats = synth(note_sequence, sample_rate=SAMPLE_RATE)
   note_plot = note_seq.plot_sequence(note_sequence, False)
   html = file_html(note_plot, CDN)
-  return (SAMPLE_RATE, array_of_floats), html
+  
+  scipy.io.wavfile.write("tmp_output.wav", SAMPLE_RATE, array_of_floats)
+  return "tmp_output.wav", html
   
 title = "Music generation with GPT-2"