Spaces:

Darkhan
/

shad_ml2

Runtime error

Darkhan commited on Mar 27, 2022

Commit

5c59066

•

1 Parent(s): 18974eb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ def classify(text, tokenizer, model):
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, num_labels=8)
-model.load_state_dict(torch.load(MODEL_PATH), map_location=torch.device('cpu'))
 model.eval()
 st.markdown("## Article classifier")

 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, num_labels=8)
+model.load_state_dict(torch.load(MODEL_PATH, map_location=torch.device('cpu')))
 model.eval()
 st.markdown("## Article classifier")