Update app.py

app.py

@@ -47,9 +47,14 @@ def proc_pil_img(input_image):
  output_image = PIL.Image.fromarray(output_image)
  return output_image
 
-def process(im):
- im = align_face(im, output_size=image_size, transform_size=transform_size)
+def process(im, is_align):
+ if is_align == 'True':
+ im = align_face(im, output_size=image_size, transform_size=transform_size)
+ else: 
+ pass
+ 
  res = proc_pil_img(im)
+ 
  return res
  
 gr.Interface(
@@ -59,7 +64,9 @@ gr.Interface(
  title="Arcane Style Transfer",
  description="Gradio demo for Arcane Style Transfer",
  article = "<p style='text-align: center'><a href='https://github.com/jjeamin/anime_style_transfer_pytorch' target='_blank'>Github Repo by jjeamin</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=jjeamin_arcane_st' alt='visitor badge'></center></p>",
- examples=[['billie.png'], ['gongyoo.jpeg'], ['IU.png'], ['elon.png']],
+ examples=[['billie.png', 'True'], ['gongyoo.jpeg', 'True'], ['IU.png', 'True'], ['elon.png', 'True']],
+ gr.Interface(inference, [gr.inputs.Image(type="pil"),
+ gr.inputs.Radio(['True','False'], type="value", default='True', label='face align')], 
  enable_queue=True,
  allow_flagging=False,
  allow_screenshot=False