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}
|