Exif transform

app.py

@@ -81,8 +81,8 @@ def inference(img):
       
   
 title = "Apocalyptify"
-description = "How will your face look after the Apocalypsehas come? Upload an image with a face, or click one of the examples below. If a face could not be detected, the output box will remain empty."
+description = "How will your face look after the Apocalypse? Upload an image with a face, or click one of the examples below. If a face could not be detected, the output box will remain empty."
 article = "<p style='text-align: center'><a href='https://github.com/justinpinkney/pixel2style2pixel/tree/nw' target='_blank'>Github Repo</a></p><p style='text-align: center'>samples: <img src='https://hf.space/gradioiframe/Norod78/Apocalyptify/file/Sample00001.jpg' alt='Sample00001'/><img src='https://hf.space/gradioiframe/Norod78/Apocalyptify/file/Sample00002.jpg' alt='Sample00002'/><img src='https://hf.space/gradioiframe/Norod78/Apocalyptify/file/Sample00003.jpg' alt='Sample00003'/><img src='https://hf.space/gradioiframe/Norod78/Apocalyptify/file/Sample00004.jpg' alt='Sample00004'/><img src='https://hf.space/gradioiframe/Norod78/Apocalyptify/file/Sample00005.jpg' alt='Sample00005'/></p><p>The Apocalypse model was fine tuned on a pre-trained Pixel2Style2Pixel model by Doron Adler</p>"
 
-examples=[['Example00001.jpg'],['Example00002.jpg'],['Example00003.jpg'],['Example00004.jpg'],['Example00005.jpg']]
+examples=[['Example00001.jpg'],['Example00002.jpg'],['Example00003.jpg'],['Example00004.jpg'],['Example00005.jpg'], ['Example00006.jpg']]
 gr.Interface(inference, gr.inputs.Image(type="pil",shape=(256,256)), gr.outputs.Image(type="pil"),title=title,description=description,article=article,examples=examples,enable_queue=True).launch()

face_detection.py

@@ -14,6 +14,11 @@ detector = dlib.get_frontal_face_detector()
 
 
 def align(image_in, face_index=0, output_size=256):
+    try:
+        image_in = ImageOps.exif_transpose(image_in)
+    except:
+        print("exif problem, not rotating")
+
     landmarks = list(get_landmarks(image_in))
     n_faces = len(landmarks)
     face_index = min(n_faces-1, face_index)
@@ -85,12 +90,7 @@ def image_align(src_img, face_landmarks, output_size=512, transform_size=2048, e
         c = eye_avg + eye_to_mouth * em_scale
         quad = np.stack([c - x - y, c - x + y, c + x + y, c + x - y])
         quad_orig = quad.copy()
-        qsize = np.hypot(*x) * 2
-
-        try:
-            src_img = ImageOps.exif_transpose(src_img)
-        except:
-            print("exif problem, not rotating")
+        qsize = np.hypot(*x) * 2        
         
         img = src_img.convert('RGBA').convert('RGB')