geraldadli commited on
Commit
2cd81df
·
verified ·
1 Parent(s): 13e5d57

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -83,7 +83,7 @@ with torch.no_grad():
83
  probs = torch.softmax(model(**inputs).logits, dim=-1)[0]
84
 
85
  print({model.config.id2label[i]: round(p.item(), 3) for i, p in enumerate(probs)})
86
- '''
87
 
88
  Labels
89
  id label
 
83
  probs = torch.softmax(model(**inputs).logits, dim=-1)[0]
84
 
85
  print({model.config.id2label[i]: round(p.item(), 3) for i, p in enumerate(probs)})
86
+ ```
87
 
88
  Labels
89
  id label