Spaces:

hunkim
/

DialoGPT

Runtime error

hunkim commited on Dec 18, 2021

Commit

5f84001

1 Parent(s): c716dd8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 @st.cache(hash_funcs=
     {transformers.models.gpt2.tokenization_gpt2_fast.GPT2TokenizerFast: hash},
-    suppress_st_warning=True)
 def load_data():
     tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")

 @st.cache(hash_funcs=
     {transformers.models.gpt2.tokenization_gpt2_fast.GPT2TokenizerFast: hash},
+    suppress_st_warning=True, allow_output_mutation=True)
 def load_data():
     tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")