Update app.py

app.py

@@ -3,14 +3,14 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from transformers import pipeline
 
-@st.cache(allow_output_mutation=True)
+@st.cache_data(allow_output_mutation=True)
 def load_model():
   model_ckpt = "flax-community/gpt2-rap-lyric-generator"
   tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
   model = AutoModelForCausalLM.from_pretrained(model_ckpt,from_flax=True)
   return tokenizer, model
 
-@st.cache()
+@st.cache_data()
 def load_rappers():
   text_file = open("rappers.txt")
   rappers = text_file.readlines()