Spaces:

ilushado
/

article_roberta_classifier

Runtime error

ilushado commited on Apr 23, 2023

Commit

76830fa

1 Parent(s): ccda969

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ class RobertaClass(torch.nn.Module):
         output = self.classifier(pooler)
         return output
-@st.cache_resource
 def load_model():
     tokenizer = RobertaTokenizer.from_pretrained('roberta-base', truncation=True, do_lower_case=True,
                                                  vocab_file='model/vocab.json',

         output = self.classifier(pooler)
         return output
+@st.cache
 def load_model():
     tokenizer = RobertaTokenizer.from_pretrained('roberta-base', truncation=True, do_lower_case=True,
                                                  vocab_file='model/vocab.json',