heegyu commited on
Commit
98103c0
1 Parent(s): 49cddd5

st.cache()

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
 
3
- @st.cache_resource()
4
  def get_pipe():
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  model_name = "heegyu/ajoublue-gpt2-medium-summarization"
 
1
  import streamlit as st
2
 
3
+ @st.cache(allow_output_mutation=True)
4
  def get_pipe():
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  model_name = "heegyu/ajoublue-gpt2-medium-summarization"