Update README.md

README.md

@@ -71,8 +71,8 @@ with torch.no_grad():
     logits = model(**inputs).logits
 
 probs = torch.sigmoid(logits).squeeze().tolist()
-labels = [model.config.id2label[str(i)] for i in range(len(probs))]
+results = {model.config.id2label[i]: float(probs[i]) for i in range(len(probs))}
+print(results)
 
-print(list(zip(labels, probs)))