Update main.py
Browse files
main.py
CHANGED
@@ -104,7 +104,7 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
|
|
104 |
logging.error("Failed to process the image: No face detected or alignment failed.")
|
105 |
return {"error": message}
|
106 |
|
107 |
-
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='
|
108 |
if processed_image is None:
|
109 |
logging.error("Failed to toonify the image.")
|
110 |
return {"error": message}
|
|
|
104 |
logging.error("Failed to process the image: No face detected or alignment failed.")
|
105 |
return {"error": message}
|
106 |
|
107 |
+
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon2-d')
|
108 |
if processed_image is None:
|
109 |
logging.error("Failed to toonify the image.")
|
110 |
return {"error": message}
|