Update app.py

app.py

@@ -17,13 +17,19 @@ import base64
 from io import BytesIO
 import os
 from PIL import Image
-import gdown
 import face_recognition
+import whatimage
+import pyheif
 
-def download_gdrive_url():
-  url = 'https://drive.google.com/u/0/uc?id=1PPO2MCttsmSqyB-vKh5C7SumwFKuhgyj&export=download'
-  output = 'haarcascade_frontalface_default.xml'
-  gdown.download(url, output, quiet=False)
+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")
   
 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()
@@ -97,6 +103,7 @@ def preprocess_mask(mask):
 # output the sampled images
 def main(user_image):
   # accept the image as input
+  decodeImage(data, user_image.name)
   init_image = PIL.Image.open(user_image).convert("RGB")
   # # resize the image to be (512, 512)
   newsize = (512, 512)