Update README.md
Browse files
README.md
CHANGED
@@ -64,6 +64,7 @@ prediction = torch.nn.functional.interpolate(
|
|
64 |
mode="bicubic",
|
65 |
align_corners=False,
|
66 |
)
|
|
|
67 |
|
68 |
# visualize the prediction
|
69 |
output = prediction.squeeze().cpu().numpy()
|
|
|
64 |
mode="bicubic",
|
65 |
align_corners=False,
|
66 |
)
|
67 |
+
```
|
68 |
|
69 |
# visualize the prediction
|
70 |
output = prediction.squeeze().cpu().numpy()
|