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