Update face alignment

app.py

@@ -1,12 +1,10 @@
 import os
-from huggingface_hub import hf_hub_download
-os.system("pip -qq install facenet_pytorch")
-from facenet_pytorch import MTCNN
-from torchvision import transforms
 import torch, PIL
 import gradio as gr
 import torch
 from utils import align_face
+from torchvision import transforms
+from huggingface_hub import hf_hub_download
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
@@ -35,9 +33,6 @@ def tensor2im(var):
      return var.mul(t_stds).add(t_means).mul(255.).clamp(0,255).permute(1,2,0)
 
 def proc_pil_img(input_image):
-    # input_image = PIL
-    input_image = align_face(input_image, output_size=image_size, transform_size=transform_size)
-    
     if 'cuda' in device: 
         transformed_image = img_transforms(input_image)[None,...].cuda().half()
     else:
@@ -51,6 +46,7 @@ def proc_pil_img(input_image):
     return output_image
 
 def process(im):
+    im = align_face(im, output_size=image_size, transform_size=transform_size)
     res = proc_pil_img(im)
     return res