Update README.md

README.md

@@ -58,16 +58,17 @@ import torch
 from PIL import Image
 from diffusers import AutoencoderKL
 from huggingface_hub import hf_hub_download
-from torchvision.transforms.functional import to_tensor, to_pil_image
+from torchvision.transforms.functional import to_tensor, to_pil_image, normalize
 
 # Load the pre-trained Emuru VAE from Hugging Face Hub.
 model = AutoencoderKL.from_pretrained("blowing-up-groundhogs/emuru_vae")
 
 # Function to preprocess an RGB image:
-# Loads the image, converts it to RGB, and transforms it to a tensor normalized to [0, 1].
+# Loads the image, converts it to RGB, and transforms it to a tensor normalized to [-1, 1].
 def preprocess_image(image_path):
     image = Image.open(image_path).convert("RGB")
     image_tensor = to_tensor(image).unsqueeze(0)  # Add batch dimension
+    image_tensor = normalize(image_tensor, [0.5], [0.5])
     return image_tensor
 
 # Function to postprocess a tensor back to a PIL image for visualization: