Update app.py

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelWithLMHead
   
 tokenizer = AutoTokenizer.from_pretrained("gpt2-large")
-@st.cache
+@st.cache_data
 def load_model(model_name):
     model = AutoModelWithLMHead.from_pretrained("gpt2-large")
     return model