Luisgust commited on
Commit
cf8a2af
·
verified ·
1 Parent(s): 2a31189

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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='cartoon1')
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}