Spaces:

flax-community
/

gpt2-indonesian-test

Runtime error

Galuh Sahid commited on Aug 1, 2021

Commit

6dd057c

•

1 Parent(s): 52b2b3c

caching

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ MODELS = {
 headers = {}
-@st.cache(show_spinner=False, persist=True, hash_funcs={transformers.models.gpt2.modeling_gpt2.GPT2LMHeadModel: lambda _: None})
 def load_gpt(model_type):
     model = GPT2LMHeadModel.from_pretrained(MODELS[model_type])

 headers = {}
+@st.cache(show_spinner=False, persist=True)
 def load_gpt(model_type):
     model = GPT2LMHeadModel.from_pretrained(MODELS[model_type])