Update README.md

README.md

@@ -74,9 +74,10 @@ def preprocess_image(image_path):
     return image_tensor
 
 # Function to postprocess a tensor back to a PIL image for visualization:
-# Clamps the tensor to [0, 1] and converts it to a PIL image.
+# Clamps the tensor to [-1, 1] and converts it to a PIL image.
 def postprocess_tensor(tensor):
-    tensor = torch.clamp(tensor, 0, 1).squeeze(0)  # Remove batch dimension
+    tensor = torch.clamp(tensor, -1, 1).squeeze(0)  # Remove batch dimension
+    tensor = (tensor + 1) / 2
     return to_pil_image(tensor)
 
 # Example: Encode and decode an image.