Update app.py

app.py

@@ -46,11 +46,11 @@ def color_dataframe(row):
             styles.append(f"background-color: None") 
     return styles
 
-@st.cache_data(allow_output_mutation=True)
+@st.cache_resource
 def load_tokenizer(model_ckpt):
     return AutoTokenizer.from_pretrained(model_ckpt)
 
-@st.cache_data(allow_output_mutation=True)
+@st.cache_resource
 def load_model(model_ckpt):
     model = AutoModelForCausalLM.from_pretrained(model_ckpt)
     return model