Update main.py
Browse files
main.py
CHANGED
@@ -222,7 +222,7 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
|
|
222 |
|
223 |
# Process the uploaded image
|
224 |
aligned_face, instyle, message = model.detect_and_align_image("uploaded_image.jpg", top, bottom, left, right)
|
225 |
-
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='
|
226 |
|
227 |
# Convert processed image to bytes
|
228 |
image_bytes = cv2.imencode('.jpg', processed_image)[1].tobytes()
|
|
|
222 |
|
223 |
# Process the uploaded image
|
224 |
aligned_face, instyle, message = model.detect_and_align_image("uploaded_image.jpg", top, bottom, left, right)
|
225 |
+
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='illustration1-d')
|
226 |
|
227 |
# Convert processed image to bytes
|
228 |
image_bytes = cv2.imencode('.jpg', processed_image)[1].tobytes()
|