Update app.py

app.py

@@ -92,6 +92,7 @@ def get_prompt(mask_token):
 
 if __name__ == "__main__":
     model_id,mask_token = get_model_id()
+    pipe = pipeline("fill-mask", model=model_id, token=hf_token)
     if st.checkbox("Show example"):
         show_example()
     sentence = get_input_text()
@@ -99,7 +100,7 @@ if __name__ == "__main__":
     if sentence:
         prompt = get_prompt(mask_token)
         if prompt and st.button("Classify"):
-            pipe = pipeline("fill-mask", model=model_id, token=hf_token)
+            # pipe = pipeline("fill-mask", model=model_id, token=hf_token)
             output = pipe(sentence + prompt, targets=TARGETS, top_k =len(TARGETS))
             output = pd.DataFrame(output)
             output['score'] = output['score'].apply(lambda x:x/sum(output['score']))