Spaces:

ki33elev
/

Transformer_arxiv_classification

Runtime error

ki33elev commited on Mar 25, 2022

Commit

c24d1f1

•

1 Parent(s): c4fdf30

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,9 @@
 import streamlit as st
 import torch
 import transformers
-@st.cache(suppress_st_warning=True, hash_funcs={transformers.AutoTokenizer: lambda _: None})
 def load_model():
     from transformers import AutoTokenizer, AutoModelForSequenceClassification
     model_name = 'distilbert-base-cased'

 import streamlit as st
 import torch
 import transformers
+import tokenizers
+@st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: lambda _: None})
 def load_model():
     from transformers import AutoTokenizer, AutoModelForSequenceClassification
     model_name = 'distilbert-base-cased'