Update main.py

main.py

@@ -17,8 +17,16 @@ sys.path.insert(0, 'vtoonify')
 
 from util import load_psp_standalone, get_video_crop_parameter, tensor2cv2
 import torch
+import torch.nn as nn
+import numpy as np
+import dlib
+import cv2
 from model.vtoonify import VToonify
 from model.bisenet.model import BiSeNet
+import torch.nn.functional as F
+from torchvision import transforms
+from model.encoder.align_all_parallel import align_face
+import gc
 import huggingface_hub
 import os
 from io import BytesIO