fix mode

gradio_painttransformer.py

@@ -12,6 +12,7 @@ import network
 from time import time
 from glob import glob
 from loguru import logger
+import numpy as np
 import gradio as gr
 
 import paddle
@@ -65,16 +66,16 @@ def predict(image_file):
     frames = []
     for idx, frame in enumerate(final_result_list):
         # out.write(frame)
-        frames.append(frame)
+        frames.append(frame[:,:,::-1]) # BGR -> RGB
     # out.release()
-    imageio.mimsave('output.gif', frames, mode="RGBA")
+    imageio.mimsave('output.gif', frames)
     logger.info('--- animation generated')
     return 'output.gif'
 
 iface = gr.Interface(
     predict,
     title='🎨 Paint Transformer',
-    description='This demo converts an image into a sequence of painted images (takes about 90 sec)',
+    description='This demo converts an image into a sequence of painted images (takes about 3 min ^^;)',
     inputs=[
         gr.inputs.Image(label='Input image', type='filepath')
     ],