Update app.py

app.py

@@ -16,7 +16,7 @@ import random
 import base64
 from io import BytesIO
 import os
-from PIL import Image
+from PIL import Image, ImageOps
 import face_recognition
 import pillow_heif
   
@@ -94,6 +94,11 @@ def main(user_image):
   # accept the image as input
   pillow_heif.register_heif_opener()
   init_image = PIL.Image.open(user_image).convert("RGB")
+  try:
+    #rotate accordingly
+    init_image = ImageOps.exif_transpose(init_image)   
+  except:
+    pass
   # # resize the image to be (512, 512)
   newsize = (512, 512)
   init_image = init_image.resize(newsize)