add initial plot

app.py

@@ -1,4 +1,5 @@
 import numpy as np
+import pickle
 from transformers import pipeline
 from transformers import AutoModelForMaskedLM, AutoTokenizer
 
@@ -111,10 +112,10 @@ with gr.Blocks() as demo:
         generate_btn = gr.Button("Generate Plot")
         mask_btn = gr.Button("Add <mask>")
 
-    plot = gr.Plot()
+    plot_example = pickle.load(open("plot_example.p", "rb"))
+    plot = gr.Plot(plot_example)
     
     #textbox.change(fn=plot_time, inputs=textbox, outputs=plot)
-    #textbox.click(fn=plot_time, inputs=textbox, outputs=plot)
     generate_btn.click(fn=plot_time, inputs=textbox, outputs=plot)
     mask_btn.click(fn=add_mask, inputs=textbox, outputs=textbox)