Update app.py

app.py

@@ -28,15 +28,15 @@ class Generator(nn.Module):
         return output
 
 
-def display_gif(file_name):
+def display_gif(file_name, save_name):
     images = []
 
     for frame in range(8):
-        frame_name = '_%d' % (frame)
+        frame_name = '%d' % (frame)
         image_filename = file_name + frame_name + '.png'
         images.append(imageio.imread(image_filename))
 
-    gif_filename = 'avatar.gif'
+    gif_filename = 'avatar_source.gif'
     return imageio.mimsave(gif_filename, images)
 
 
@@ -72,10 +72,10 @@ def run(action, body, hair, top, bottom):
     elif bottom == "red": bottom = '2'
     elif bottom == "silver": bottom = '3'
     
-    name = './Sprite/frames/domain_1/' + action + '/'
-    name = name + 'front' + '_' + str(body) + str(bottom) + str(top) + str(hair) + '_'
+    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_image(name)
+    gif = display_gif(file_name_source, 'avatar_source.gif')
     
     return 'avatar_source.gif'