Ashrafb commited on
Commit
ef2a14d
·
verified ·
1 Parent(s): ebb8d7c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -1
main.py CHANGED
@@ -10,7 +10,13 @@ import dlib
10
  from torchvision import transforms
11
  import torch.nn.functional as F
12
  from vtoonify_model import Model # Importing the Model class from vtoonify_model.py
13
- from util import load_psp_standalone, get_video_crop_parameter, tensor2cv2, align_face
 
 
 
 
 
 
14
  from model.vtoonify import VToonify
15
  from model.bisenet.model import BiSeNet
16
  import huggingface_hub
 
10
  from torchvision import transforms
11
  import torch.nn.functional as F
12
  from vtoonify_model import Model # Importing the Model class from vtoonify_model.py
13
+ import gradio as gr
14
+ import pathlib
15
+ import sys
16
+ sys.path.insert(0, 'vtoonify')
17
+
18
+ from util import load_psp_standalone, get_video_crop_parameter, tensor2cv2
19
+ import torch
20
  from model.vtoonify import VToonify
21
  from model.bisenet.model import BiSeNet
22
  import huggingface_hub