Update app.py

app.py

@@ -6,7 +6,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 checkpoint = "."
   
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-@st.cache
+
+@st.cache_resource
 def load_model(model_name):
     model = AutoModelForCausalLM.from_pretrained(model_name)
     return model