Spaces:

ki33elev
/

Transformer_arxiv_classification

Runtime error

ki33elev commited on Mar 25, 2022

Commit

0fbdf0a

•

1 Parent(s): 6a12a2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def load_model():
     model.eval()
     return tokenizer, model
-@st.cache(suppress_st_warning=True)
 def predict(title, summary, tokenizer, model):
     text = title + "\n" + summary
     tokens = tokenizer.encode(text)

     model.eval()
     return tokenizer, model
+@st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: lambda _: None})
 def predict(title, summary, tokenizer, model):
     text = title + "\n" + summary
     tokens = tokenizer.encode(text)