Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ import random
|
|
16 |
import base64
|
17 |
from io import BytesIO
|
18 |
import os
|
19 |
-
from PIL import Image
|
20 |
import face_recognition
|
21 |
import pillow_heif
|
22 |
|
@@ -94,6 +94,11 @@ def main(user_image):
|
|
94 |
# accept the image as input
|
95 |
pillow_heif.register_heif_opener()
|
96 |
init_image = PIL.Image.open(user_image).convert("RGB")
|
|
|
|
|
|
|
|
|
|
|
97 |
# # resize the image to be (512, 512)
|
98 |
newsize = (512, 512)
|
99 |
init_image = init_image.resize(newsize)
|
|
|
16 |
import base64
|
17 |
from io import BytesIO
|
18 |
import os
|
19 |
+
from PIL import Image, ImageOps
|
20 |
import face_recognition
|
21 |
import pillow_heif
|
22 |
|
|
|
94 |
# accept the image as input
|
95 |
pillow_heif.register_heif_opener()
|
96 |
init_image = PIL.Image.open(user_image).convert("RGB")
|
97 |
+
try:
|
98 |
+
#rotate accordingly
|
99 |
+
init_image = ImageOps.exif_transpose(init_image)
|
100 |
+
except:
|
101 |
+
pass
|
102 |
# # resize the image to be (512, 512)
|
103 |
newsize = (512, 512)
|
104 |
init_image = init_image.resize(newsize)
|