Update app.py

app.py

@@ -39,13 +39,11 @@ def run(search_prompt, mid=None, progress=gr.Progress()):
         x.append(s[1] / mid_seq_ticks)
         y.append(s[4])
         c.append(colors[s[3]])
-
-    plot = gr.ScatterPlot(x=x, y=y, color=c)
  
     with open(f"output.mid", 'wb') as f:
         f.write(MIDI.score2midi([mid_seq_ticks, mid_seq]))
     audio = synthesis(MIDI.score2opus([mid_seq_ticks, mid_seq]), soundfont_path)
-    yield mdata, "output.mid", (44100, audio), plot
+    yield mdata, "output.mid", (44100, audio), x, y, c
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()