Update app.py

app.py

@@ -510,7 +510,7 @@ def main(input_path, model_path, output_dir, need_animation=False, resize_h=None
 
         all_frames[0].save(os.path.join(frame_dir, 'animation.gif'),
                save_all=True, append_images=all_frames[1:], optimize=False, duration=40, loop=0)
-        return os.path.join(frame_dir, "animation.gif")
+        return os.path.join(frame_dir, "animation.gif"), tensor_to_pil
 
 
 def gradio_inference(image):
@@ -518,8 +518,8 @@ def gradio_inference(image):
          model_path='model.pth',
          output_dir='output/',
          need_animation=True,  # whether need intermediate results for animation.
-         resize_h=400,         # resize original input to this size. None means do not resize.
-         resize_w=400,         # resize original input to this size. None means do not resize.
+         resize_h=300,         # resize original input to this size. None means do not resize.
+         resize_w=300,         # resize original input to this size. None means do not resize.
          serial=True)          # if need animation, serial must be True.
 
 title = "Paint Transformer"
@@ -528,8 +528,9 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.037
 
 gr.Interface(
     gradio_inference, 
-    [gr.inputs.Image(type="file", label="Input")], 
-    gr.outputs.Image(type="file", label="Output"),
+    gr.inputs.Image(type="file", label="Input"), 
+    [gr.outputs.Image(type="file", label="Output GIF"),
+    gr.outputs.Image(type="pil", label="Output Image")],
     title=title,
     description=description,
     article=article,