Update main.py
Browse files
main.py
CHANGED
@@ -34,7 +34,7 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
|
|
34 |
shutil.copyfileobj(file.file, buffer)
|
35 |
|
36 |
# Process the uploaded image
|
37 |
-
aligned_face, instyle, message = model.detect_and_align_image("uploaded_image.jpg", top, bottom, left, right)
|
38 |
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon1')
|
39 |
|
40 |
# Convert processed image to bytes
|
|
|
34 |
shutil.copyfileobj(file.file, buffer)
|
35 |
|
36 |
# Process the uploaded image
|
37 |
+
aligned_face, instyle, message = model.detect_and_align_image(self,"uploaded_image.jpg", top, bottom, left, right)
|
38 |
processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon1')
|
39 |
|
40 |
# Convert processed image to bytes
|