Update main.py

main.py

@@ -204,7 +204,6 @@ class Model:
 
 model = Model(device='cuda' if torch.cuda.is_available() else 'cpu')
 
-
 @app.post("/upload/")
 async def process_image(file: UploadFile = File(...), top: int = Form(...), bottom: int = Form(...), left: int = Form(...), right: int = Form(...)):
     if model is None:
@@ -218,7 +217,7 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
     aligned_face, instyle, message = model.detect_and_align_image("uploaded_image.jpg", top, bottom, left, right)
     instyle_np = instyle['instyle']  # Extract the NumPy array from the dictionary
 
-    processed_image, message = model.image_toonify(aligned_face, instyle_np, model.exstyle, style_degree=0.5, style_type='illustration1-d')
+    processed_image, message = model.image_toonify(aligned_face, instyle_np, model.exstyle, style_degree=0.5, style_type='cartoon3-d')
 
     # Convert processed image to bytes
     image_bytes = cv2.imencode('.jpg', processed_image)[1].tobytes()