Update README.md

README.md

@@ -64,13 +64,13 @@ prediction = torch.nn.functional.interpolate(
     mode="bicubic",
     align_corners=False,
 )
-```
 
 # visualize the prediction
 output = prediction.squeeze().cpu().numpy()
 formatted = (output * 255 / np.max(output)).astype("uint8")
 depth = Image.fromarray(formatted)
 depth.show()
+```
 
 For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/dpt).