Spaces:

heegyu
/

ajoublue-gpt2-medium-summarization

Runtime error

heegyu commited on Mar 5, 2023

Commit

98103c0

1 Parent(s): 49cddd5

st.cache()

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import streamlit as st
-@st.cache_resource()
 def get_pipe():
     from transformers import AutoTokenizer, AutoModelForCausalLM
     model_name = "heegyu/ajoublue-gpt2-medium-summarization"

 import streamlit as st
+@st.cache(allow_output_mutation=True)
 def get_pipe():
     from transformers import AutoTokenizer, AutoModelForCausalLM
     model_name = "heegyu/ajoublue-gpt2-medium-summarization"