Update README.md
Browse files
README.md
CHANGED
|
@@ -57,7 +57,7 @@ model = model.from_pretrained(repo_name)
|
|
| 57 |
if cuda_flag:
|
| 58 |
model.cuda()
|
| 59 |
|
| 60 |
-
|
| 61 |
sr_results = model.infer_image("path-to-input-image", cuda=cuda_flag)
|
| 62 |
cv2.imwrite("path-to-output-location", sr_results)
|
| 63 |
```
|
|
|
|
| 57 |
if cuda_flag:
|
| 58 |
model.cuda()
|
| 59 |
|
| 60 |
+
# test and save results
|
| 61 |
sr_results = model.infer_image("path-to-input-image", cuda=cuda_flag)
|
| 62 |
cv2.imwrite("path-to-output-location", sr_results)
|
| 63 |
```
|