fix pickle load (plot)

app.py

@@ -1,17 +1,19 @@
 import numpy as np
 import pickle
+import urllib
 from transformers import pipeline
 from transformers import AutoModelForMaskedLM, AutoTokenizer
 
 import gradio as gr
 import matplotlib.pyplot as plt
 
+plot_url = "https://huggingface.co/spaces/fvancesco/test_time_1.1/resolve/main/plot_example.p"
+
 dates = []
 dates.extend([f"18 {m}" for m in range(1,13)])
 dates.extend([f"19 {m}" for m in range(1,13)])
 dates.extend([f"20 {m}" for m in range(1,13)])
 dates.extend([f"21 {m}" for m in range(1,13)])
-
 months = [x.split(" ")[-1] for x in dates]
 
 model_name = "fvancesco/tmp_date"
@@ -21,8 +23,6 @@ model.eval()
 #pipe = pipeline('fill-mask', model=model, tokenizer=tokenizer, device=0)
 pipe = pipeline('fill-mask', model=model, tokenizer=tokenizer)
 
-last_mf_dict = None
-
 
 def get_mf_dict(text):
 
@@ -58,7 +58,6 @@ def get_mf_dict(text):
 
 def plot_time(text):
     mf_dict = get_mf_dict(text)
-    #last_mf_dict = mf_dict # just for debugging, remove in final version
     
     #max_tokens = 10
 
@@ -105,14 +104,16 @@ def add_mask(text):
     return out
 
 with gr.Blocks() as demo:
-    #textbox = gr.Textbox(placeholder="Type here and press enter...")
+    
     textbox = gr.Textbox(value="Happy <mask>!", max_lines=1)
 
     with gr.Row():
         generate_btn = gr.Button("Generate Plot")
         mask_btn = gr.Button("Add <mask>")
 
-    plot_example = pickle.load(open("plot_example.p", "rb"))
+    # plot (with starting example already loaded)
+    f = urllib.request.urlopen(plot_url)
+    plot_example = pickle.load(f)
     plot = gr.Plot(plot_example)
     
     #textbox.change(fn=plot_time, inputs=textbox, outputs=plot)