Update app.py

app.py

@@ -1,6 +1,7 @@
+import gradio as gr
 import torch
 from torch import nn
-import gradio as gr
+import imageio
 
 
 class Generator(nn.Module):
@@ -27,6 +28,18 @@ class Generator(nn.Module):
         return output
 
 
+def display_gif(file_name):
+    images = []
+
+    for frame in range(8):
+        frame_name = '_%d' % (frame)
+        image_filename = file_name + frame_name + '.png'
+        images.append(imageio.imread(image_filename))
+
+    gif_filename = 'avatar.gif'
+    return imageio.mimsave(gif_filename, images)
+
+
 def path(action, body, hair, top, bottom):
 
     # body
@@ -50,9 +63,13 @@ def path(action, body, hair, top, bottom):
     elif bottom == "golden": bottom = '1'
     elif bottom == "red": bottom = '2'
     elif bottom == "silver": bottom = '3'
-
-    name = 'front' + '_' + str(body) + str(bottom) + str(top) + str(hair)
-    return name
+    
+    name = 'Sprite/' + 'domain_1/' + action + '/'
+    name = name + 'front' + '_' + str(body) + str(bottom) + str(top) + str(hair) + '_'
+    
+    gif = display_gif(name)
+    
+    return gif
 
 
 gr.Interface(