Update app.py

app.py

@@ -151,8 +151,8 @@ def GenerateMIDI(num_tok, idrums, iinstr, progress=gr.Progress()):
       y.append(s[4])
       c.append(colors[s[3]])
 
-    fig = plt.figure(figsize=(14,5))
-    midi_plot = fig.axes(title='Allegro Music Transformer Composition')
+    midi_plot = plt.figure(figsize=(14,5))
+    midi_plot = plt.axes(title='Allegro Music Transformer Composition')
     midi_plot = ax.set_facecolor('black')
     plt.scatter(x,y, c=c)
     plt.xlabel("Time")