Also Exif transpose images without detected faces

.gitignore

@@ -0,0 +1,2 @@
+*.pyc
+gradio_queue.db

app.py

@@ -1,6 +1,5 @@
 import os
-#os.system("gdown https://drive.google.com/uc?id=1WEST2O6svlQWpJNomX3947Q2bfJz4bAJ")
-#os.system("gdown https://drive.google.com/uc?id=1CbnhlUI9Tms2o7S2eCg9qwGXZFCyROYy")
+os.system("pip install --upgrade pip")
 os.system("pip install dlib")
 import sys
 import face_detection
@@ -36,6 +35,10 @@ def find_aligned_face(image_in, size=256):
 def align_first_face(image_in, size=256):
     aligned_image, n_faces, quad = find_aligned_face(image_in,size=size)
     if n_faces == 0:
+        try:
+            image_in = ImageOps.exif_transpose(image_in)
+        except:
+            print("exif problem, not rotating")
         image_in = image_in.resize((size, size))
         im_array = image_as_array(image_in)
     else:
@@ -57,10 +60,14 @@ def face2drag(
 ) -> Image.Image:
 
     aligned_img = align_first_face(img)
-    input = torch.Tensor(aligned_img)
-    output = net(input)
-    output = tensor2im(output[0])
-    output = img_concat_h(tensor2im(torch.Tensor(aligned_img)[0]), output)
+    if aligned_img is None:
+        output=None
+    else:
+        input = torch.Tensor(aligned_img)
+        output = net(input)
+        output = tensor2im(output[0])
+        output = img_concat_h(tensor2im(torch.Tensor(aligned_img)[0]), output)
+
     return output
     
 import os