Update README.md

2022-12-07 12:50:02 +00:00 · 2022-12-07 12:50:02 +00:00 · 402bc4afca
parent 25883a7ba6
commit 402bc4afca
1 changed files with 1 additions and 0 deletions
--- 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()