Spaces:

rrevoid
/

article_classifier

Runtime error

rrevoid commited on Mar 27, 2022

Commit

aabbe07

1 Parent(s): 17fb100

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch.nn as nn
 from transformers import RobertaTokenizer, RobertaModel
-#@st.cache(suppress_st_warning=True)
 def init_model():
     model = RobertaModel.from_pretrained("roberta-large-mnli")

 from transformers import RobertaTokenizer, RobertaModel
+@st.cache(suppress_st_warning=True)
 def init_model():
     model = RobertaModel.from_pretrained("roberta-large-mnli")