Update app.py

app.py

@@ -165,15 +165,15 @@ def GenerateMIDI(progress=gr.Progress()):
       y.append(s[4])
       c.append(colors[s[3]])
 
-    fig = plt.figure(figsize=(14,5))
-    ax=plt.axes(title=fname)
+    plt.figure(figsize=(14,5))
+    ax=plt.axes(title='Allegro Music Transformer')
     ax.set_facecolor('black')
     
     plt.scatter(x,y, c=c)
     plt.xlabel("Time")
     plt.ylabel("Pitch")
 
-    yield [500, output1], fig, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)
+    yield [500, output1], plt, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)
         
 #=================================================================================================