update readme

README.md

@@ -145,6 +145,7 @@ with torch.cuda.amp.autocast(dtype=torch.bfloat16):
 prediction = tokenizer.decode(generated_text[0], skip_special_tokens=True)
 print("==> prediciton: ", prediction)
 print("-"*120)
+# ==> prediciton:  A man sitting on a bench in front of a red building.
 ```
 
 More comprehensive examples can be found in the [notebook](demo.ipynb).