Update README.md

README.md

@@ -66,15 +66,13 @@ with  torch.no_grad():
 	logits  =  model_output.logits.squeeze(-1).float().cpu().numpy()
 
 # scores are produced in the range [0, 4]. To convert to the range [1, 5], we can simply add 1 to the score.
-score = logits.item() + 1 
+score = [x + 1 for x in logits.tolist()][0]
 
-result = {
+print({
  "text": text,
  "score": score,
- "toxic_score": int(round(max(0, min(score, 4)))) + 1,
-}
-
-print(result)
+ "int_score": [int(round(max(0, min(score, 4)))) + 1 for score in logits][0],
+})
 ```