Update app.py

app.py

@@ -151,6 +151,7 @@ def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
       y.append(s[4])
       c.append(colors[s[3]])
 
+    plt.close()
     plt.figure(figsize=(14,5))
     ax=plt.axes(title='Allegro Music Transformer Composition')
     ax.set_facecolor('black')
@@ -158,8 +159,7 @@ def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
     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)
         
 #=================================================================================================