Hugging Face's logo

Spaces:
Darkhan
/
shad_ml2
Runtime error

App Files Community
Darkhan commited on
Commit
49d69f1
•
1 Parent(s): 33b111e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def classify(text, tokenizer, model):
35
 
36
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
37
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, num_labels=8)
38
- # model.load_state_dict(torch.load(model_path))
39
  model.eval()
40
 
41
  st.markdown("## Article classifier")
 
35
 
36
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
37
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, num_labels=8)
38
+ model.load_state_dict(torch.load(MODEL_PATH))
39
  model.eval()
40
 
41
  st.markdown("## Article classifier")