Update swin_ir_2.py

swin_ir_2.py

@@ -1322,4 +1322,5 @@ def preprocesss_image(image: PIL.Image.Image) -> torch.FloatTensor:
 def postprocess_image(output: torch.Tensor) -> PIL.Image.Image:
     output = output.data.float().cpu().clamp_(0, 1).numpy()
     output = (output * 255).round().astype("uint8")
-    output = output.transpose(1, 2, 0)
+    output = output.transpose(1, 2, 0)
+    return PIL.Image.fromarray(output)