Update app.py

app.py

@@ -19,7 +19,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 # device_map = infer_auto_device_map(model, dtype="float16")
 # st.write(device_map)
 
-@st.cache_resource
+@st.cache
 def load_model():
     st.write('Loading model...')
     model = pipeline("text-generation", model="bryanmildort/gpt_neo_notes")