Spaces:

rrevoid
/

article_classifier

Runtime error

rrevoid commited on Mar 27, 2022

Commit

6c23168

1 Parent(s): c2837fc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,8 @@ import streamlit as st
 import torch.nn as nn
 from transformers import RobertaTokenizer, RobertaModel
-@st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None})
 def init():
     tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")

 import torch.nn as nn
 from transformers import RobertaTokenizer, RobertaModel
+@st.cache(hash_funcs={tokenizers.AddedToken: lambda _: None})
 def init():
     tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")