Update app.py

app.py

@@ -18,18 +18,7 @@ from io import BytesIO
 import os
 from PIL import Image
 import face_recognition
-import whatimage
-import pyheif
-
-def decodeImage(bytesIo, filename):
-  fmt = whatimage.identify_image(bytesIo)
-  print(fmt)
-  if fmt in ['heic', 'avif']:
-    i = pyheif.read_heif(bytesIo)
-    pi = Image.frombytes(
-          mode=i.mode, size=i.size, data=i.data)
-    print(pi)
-    pi.save(filename, format="jpeg")
+import pillow_heif
   
 def inpaint(p, init_image, mask_image=None, strength=0.75, guidance_scale=7.5, generator=None, num_samples=1, n_iter=1):
     buffered_init_img = BytesIO()
@@ -103,9 +92,7 @@ def preprocess_mask(mask):
 # output the sampled images
 def main(user_image):
   # accept the image as input
-  with open(user_image.name, 'rb') as f:
-    data = f.read()
-  decodeImage(data, user_image.name)
+  pillow_heif.register_heif_opener()
   init_image = PIL.Image.open(user_image).convert("RGB")
   # # resize the image to be (512, 512)
   newsize = (512, 512)