dfucci commited on
Commit
28dcf06
1 Parent(s): 2bb477f

update usage example

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -34,7 +34,7 @@ with torch.no_grad():
34
  probs = model(**input).logits
35
 
36
  predicted_label = torch.argmax(probs, dim=1).item()
37
- print(predicted_label) # 0 is neutral, 1 is gendered
38
  ```
39
 
40
  ## Citation
 
34
  probs = model(**input).logits
35
 
36
  predicted_label = torch.argmax(probs, dim=1).item()
37
+ print(model.config.id2label[predicted_label])
38
  ```
39
 
40
  ## Citation