diff --git a/README.md b/README.md
index 31b888a..a8e9c1b 100644
--- a/README.md
+++ b/README.md
@@ -64,6 +64,7 @@ prediction = torch.nn.functional.interpolate(
     mode="bicubic",
     align_corners=False,
 )
+```
 
 # visualize the prediction
 output = prediction.squeeze().cpu().numpy()