removed squeeze and permute ops on sr imagery. letting the user handle that

Author: isaaccorley

torch_enhance/models/base.py

@@ -79,6 +79,4 @@ def enhance(self, x: torch.Tensor) -> torch.Tensor:
         x *= 255.0
         x = x.clamp(0, 255)
         x = x.to(torch.uint8)
-        x = x.squeeze(0)
-        x = x.permute(1, 2, 0)
         return x