Update main.py
Browse files
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='
|
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)
|