Update app.py

app.py

@@ -92,9 +92,8 @@ def preprocess_mask(mask):
 # output the sampled images
 def main(user_image):
   # accept the image as input
-  return user_image
-  pillow_heif.register_heif_opener()
-  init_image = PIL.Image.open(user_image).convert("RGB")
+  init_image = PIL.Image.open(user_image)
+  return init_image
   try:
     #rotate accordingly
     init_image = ImageOps.exif_transpose(init_image)