Spaces:

ki33elev
/

Transformer_arxiv_classification

Runtime error

ki33elev commited on Mar 25, 2022

Commit

c4fdf30

1 Parent(s): b0c8217

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 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'