Spaces:

Jipski
/

Flo-BasBoettcher

Runtime error

Jipski commited on Dec 7, 2021

Commit

db5eb1e

•

1 Parent(s): 86f1692

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import transformers
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelWithLMHead
 tokenizer = AutoTokenizer.from_pretrained("anonymous-german-nlp/german-gpt2")
-@st.cache
 def load_model(model_name):
     model = AutoModelWithLMHead.from_pretrained("Jipski/gpt2-Flo-BasBoettcher")
     return model

 import streamlit as st
 from transformers import AutoTokenizer, AutoModelWithLMHead
 tokenizer = AutoTokenizer.from_pretrained("anonymous-german-nlp/german-gpt2")
+@st.cache(allow_output_mutation=True)
 def load_model(model_name):
     model = AutoModelWithLMHead.from_pretrained("Jipski/gpt2-Flo-BasBoettcher")
     return model