Spaces:

PeterBrendan
/

pbjs_gpt2

Sleeping

PeterBrendan commited on Jul 15, 2023

Commit

28014bc

1 Parent(s): ad51725

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import pipeline
-@st.cache(allow_output_mutation=True)
 def load_model(model_name):
     return pipeline("text-generation", model=model_name)

 import streamlit as st
 from transformers import pipeline
+@st.cache_resource
 def load_model(model_name):
     return pipeline("text-generation", model=model_name)