Ashrafb commited on
Commit
aeaa85d
·
verified ·
1 Parent(s): 695a923

Update main.py

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