Fix: Correct prediction printing logic in example code

README.md

@@ -32,5 +32,5 @@ predictions = torch.round((torch.sign(discriminator_outputs[0]) + 1) / 2)
 
 [print("%7s" % token, end="") for token in fake_tokens]
 
-[print("%7s" % int(prediction), end="") for prediction in predictions.tolist()]
+[print("%7s" % int(pred), end="") for pred in predictions[0]]
 ```