Update main.py
Browse files
main.py
CHANGED
@@ -54,7 +54,7 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
|
|
54 |
logging.error("Failed to process the image: No face detected or alignment failed.")
|
55 |
return {"error": message}
|
56 |
|
57 |
-
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='
|
58 |
if processed_image is None:
|
59 |
logging.error("Failed to toonify the image.")
|
60 |
return {"error": message}
|
|
|
54 |
logging.error("Failed to process the image: No face detected or alignment failed.")
|
55 |
return {"error": message}
|
56 |
|
57 |
+
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon4')
|
58 |
if processed_image is None:
|
59 |
logging.error("Failed to toonify the image.")
|
60 |
return {"error": message}
|