Ashrafb commited on
Commit
e49db60
·
verified ·
1 Parent(s): b09de3d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -34,7 +34,7 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
34
 
35
  # Process the uploaded image
36
  aligned_face, instyle, message = model.detect_and_align_image(frame_rgb, top, bottom, left, right)
37
- processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon4')
38
 
39
  # Convert BGR to RGB
40
  processed_image_rgb = cv2.cvtColor(processed_image, cv2.COLOR_BGR2RGB)
 
34
 
35
  # Process the uploaded image
36
  aligned_face, instyle, message = model.detect_and_align_image(frame_rgb, top, bottom, left, right)
37
+ processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon1')
38
 
39
  # Convert BGR to RGB
40
  processed_image_rgb = cv2.cvtColor(processed_image, cv2.COLOR_BGR2RGB)