Update vtoonify_model.py

vtoonify_model.py

@@ -1,4 +1,10 @@
 from __future__ import annotations
+import gradio as gr
+import pathlib
+import sys
+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
@@ -8,10 +14,16 @@ 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
 import logging
 from PIL import Image
 
+
+
+
 # Configure logging
 logging.basicConfig(level=logging.INFO)