Update app.py

app.py

@@ -6,9 +6,10 @@ def context_text(text): return f"### Context\n{text}\n\n### Answer"
 
 @st.cache_resource
 def load_pipe():
-    return pipeline("token-classification", model="MSey/pbt_CaBERT_7_c10731")
+    model_name = "MSey/pbt_CaBERT_7_c10731"
+    return pipeline("token-classification", model=model_name), AutoTokenizer.from_pretrained(model_name)
 
-pipe = load_pipe()
+pipe, tokenizer = load_pipe()
 
 st.header("Test Environment for pbt_CaBERT_7_c10731")
 user_input = st.text_input("Enter your Prompt here:", "")