Ashrafb commited on
Commit
83be299
·
verified ·
1 Parent(s): 013f6fb

Update main.py

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