XiangZ
/

hit-sr

XiangZ commited on Sep 15, 2024

Commit

a57b4a9

verified ·

1 Parent(s): 0fec5d3

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -57,7 +57,7 @@ model = model.from_pretrained(repo_name)
 if cuda_flag:
     model.cuda()
-## test and save results
 sr_results = model.infer_image("path-to-input-image", cuda=cuda_flag)
 cv2.imwrite("path-to-output-location", sr_results)
 ```

 if cuda_flag:
     model.cuda()
+# test and save results
 sr_results = model.infer_image("path-to-input-image", cuda=cuda_flag)
 cv2.imwrite("path-to-output-location", sr_results)
 ```