Update app.py

app.py

@@ -152,11 +152,11 @@ def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
       c.append(colors[s[3]])
 
     midi_plot = plt.figure(figsize=(14,5))
-    midi_plot.axes(title='Allegro Music Transformer Composition')
-    midi_plot.set_facecolor('black')
-    midi_plot.scatter(x,y, c=c)
-    midi_plot.xlabel("Time")
-    midi_plot.ylabel("Pitch")
+    plt.axes(title='Allegro Music Transformer Composition')
+    plt.axes.set_facecolor('black')
+    plt.scatter(x,y, c=c)
+    plt.xlabel("Time")
+    plt.ylabel("Pitch")
     plt.close()
 
     yield [500, output1], midi_plot, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)