Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from torch import nn
 import imageio
+import cv2
 
 
 class Generator(nn.Module):
@@ -40,6 +41,21 @@ def display_gif(file_name, save_name):
     return imageio.mimsave(gif_filename, images)
 
 
+def display_gif_pad(file_name, save_name):
+    images = []
+
+    for frame in range(8):
+        frame_name = '%d' % (frame)
+        image_filename = file_name + frame_name + '.png'
+        image = imageio.imread(image_filename)
+        image = image[:, :, :3]
+        image_pad = cv2.copyMakeBorder(image, 0, 0, 32, 32, cv2.BORDER_CONSTANT, value=0)
+        images.append(image_pad)
+
+    gif_filename = 'avatar_source.gif'
+    return imageio.mimsave(gif_filename, images)
+    
+
 def display_image(file_name):
 
     image_filename = file_name + '0' + '.png'
@@ -75,7 +91,7 @@ def run(action, body, hair, top, bottom):
     file_name_source = './Sprite/frames/domain_1/' + action + '/'
     file_name_source = file_name_source + 'front' + '_' + str(body) + str(bottom) + str(top) + str(hair) + '_'
     
-    gif = display_gif(file_name_source, 'avatar_source.gif')
+    gif = display_gif_pad(file_name_source, 'avatar_source.gif')
     
     return 'avatar_source.gif'