Update app.py

app.py

@@ -115,17 +115,17 @@ def process(text):
   input_ids = tokenizer.encode(text, return_tensors="pt")
   generated_ids = model.generate(input_ids, max_length=500)
   generated_sequence = tokenizer.decode(generated_ids[0])
-  print(generated_sequence)
 
   # Convert text of notes to audio
   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)
-  note_plot = note_seq.plot_sequence(note_sequence, False)
+  note_plot = note_seq.plot_sequence(note_sequence, True)
   html = file_html(note_plot, CDN)
   
-  scipy.io.wavfile.write("tmp_output.wav", SAMPLE_RATE, array_of_floats)
-  return "tmp_output.wav", html
+  with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
+    scipy.io.wavfile.write(fp.name, SAMPLE_RATE, array_of_floats)
+    return fp.name, html
   
 title = "Music generation with GPT-2"