Update main.py

main.py

@@ -14,7 +14,7 @@ import gradio as gr
 import pathlib
 import sys
 sys.path.insert(0, 'vtoonify')
-
+from vtoonify_model import Model
 from util import load_psp_standalone, get_video_crop_parameter, tensor2cv2
 import torch
 import torch.nn as nn
@@ -191,8 +191,8 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
         shutil.copyfileobj(file.file, buffer)
 
     # Process the uploaded image
-    aligned_face, instyle, message = detect_and_align_image("uploaded_image.jpg", top, bottom, left, right)
-    processed_image, message = image_toonify(aligned_face, instyle, exstyle, style_degree=0.5, style_type='cartoon1')
+    aligned_face, instyle, message = model.detect_and_align_image("uploaded_image.jpg", top, bottom, left, right)
+    processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon1')
 
     # Convert processed image to bytes
     image_bytes = cv2.imencode('.jpg', processed_image)[1].tobytes()