Update README.md

README.md

@@ -89,8 +89,8 @@ with torch.no_grad():
 probabilities = torch.sigmoid(logits).cpu().numpy().flatten()
 
 # Format predictions
-formatted_predictions = {labels[i]: probabilities[i] for i in range(len(labels))}
-print(formatted_predictions)
+predictions = {labels[i]: probabilities[i] for i in range(len(labels))}
+print(predictions)
 ```