Spaces:

heegyu
/

ajoublue-gpt2-medium-dialog

Runtime error

heegyu commited on Mar 5, 2023

Commit

c91acd0

•

1 Parent(s): 24ea594

@st .cache(allow_output_mutation=True)

Files changed (1) hide show

app.py CHANGED Viewed

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

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