jjeamin commited on
Commit
14567ed
1 Parent(s): 8e8807d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
  os.system("pip freeze")
3
 
4
- import torch, PIL
 
5
  import gradio as gr
6
  import torch
7
  from utils import align_face
@@ -48,6 +49,8 @@ def proc_pil_img(input_image):
48
  return output_image
49
 
50
  def process(im, is_align):
 
 
51
  if is_align == 'True':
52
  im = align_face(im, output_size=image_size, transform_size=transform_size)
53
  else:
 
1
  import os
2
  os.system("pip freeze")
3
 
4
+ import torch
5
+ import PIL
6
  import gradio as gr
7
  import torch
8
  from utils import align_face
 
49
  return output_image
50
 
51
  def process(im, is_align):
52
+ im = PIL.ImageOps.exif_transpose(im)
53
+
54
  if is_align == 'True':
55
  im = align_face(im, output_size=image_size, transform_size=transform_size)
56
  else: