Update app.py

app.py

@@ -40,6 +40,14 @@ def display_gif(file_name):
     return imageio.mimsave(gif_filename, images)
 
 
+def display_image(file_name):
+
+    image_filename = file_name + '0' + '.png'
+    print(image_filename)
+    image = imageio.imread(image_filename)
+    imageio.imwrite(image, 'image.png')
+    
+
 def path(action, body, hair, top, bottom):
 
     # body
@@ -67,9 +75,9 @@ def path(action, body, hair, top, bottom):
     name = './Sprite/frames/domain_1/' + action + '/'
     name = name + 'front' + '_' + str(body) + str(bottom) + str(top) + str(hair) + '_'
     
-    gif = display_gif(name)
+    gif = display_image(name)
     
-    return 'avatar.gif'
+    return 'image.png'
 
 
 gr.Interface(