Update app.py

app.py

@@ -151,15 +151,16 @@ def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
       y.append(s[4])
       c.append(colors[s[3]])
 
-    plt.figure(figsize=(14,5))
+    midi_plot = plt.figure(figsize=(14,5))
     ax=plt.axes(title='Allegro Music Transformer Composition')
     ax.set_facecolor('black')
     
     plt.scatter(x,y, c=c)
     plt.xlabel("Time")
     plt.ylabel("Pitch")
+    plt.close()
 
-    yield [500, output1], plt, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)
+    yield [500, output1], midi_plot, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)
         
 #=================================================================================================
 
@@ -219,5 +220,5 @@ if __name__ == "__main__":
         output_plot = gr.Plot(label="output plot")
         output_midi = gr.File(label="output midi", file_types=[".mid"])
         run_event = run_btn.click(GenerateMIDI, [input_num_tokens, input_drums, input_instrument], [output_midi_seq, output_plot, output_midi, output_audio])
-        plt.close()
+
         app.queue(concurrency_count=1).launch(server_port=opt.port, share=opt.share, inbrowser=True)