Update README.md

README.md

@@ -3,6 +3,7 @@ tags:
 - HiT-SR
 - image super-resolution
 - transformer
+- efficient transformer
 ---
 
 <h1>
@@ -33,7 +34,7 @@ For each HiT-SR model, we provide 2x, 3x, 4x upscaling versions:
 | `XiangZ/hit-sng-2x` |   | HiT-SNG |   | 2x      |
 | `XiangZ/hit-sng-3x` |   | HiT-SNG |   | 3x      |
 | `XiangZ/hit-sng-4x` |   | HiT-SNG |   | 4x      |
-| `XiangZ/hit-srf-2x` |   | HiT-SNG |   | 2x      |
+| `XiangZ/hit-srf-2x` |   | HiT-SRF |   | 2x      |
 | `XiangZ/hit-srf-3x` |   | HiT-SRF |   | 3x      |
 | `XiangZ/hit-srf-4x` |   | HiT-SRF |   | 4x      |
 
@@ -57,8 +58,7 @@ if cuda_flag:
     model.cuda()
 
 ## test and save results
-image_path = "path-to-input-image"
-sr_results = model.infer_image(image_path, cuda=cuda_flag)
+sr_results = model.infer_image("path-to-input-image", cuda=cuda_flag)
 cv2.imwrite("path-to-output-location", sr_results)
 ```