replacebg / vtoonify_model.py
Munaf1987's picture
Create vtoonify_model.py
c655cdc verified
raw
history blame
242 Bytes
# vtoonify_model.py
import torch
from some_vtoonify_lib import VToonifyModel # replace with actual
def load_vtoonify():
model = VToonifyModel.from_pretrained("PKUWilliamYang/VToonify")
model.eval()
return lambda img: model(img)