Update app.py

app.py

@@ -48,7 +48,7 @@ def display_image(file_name):
     imageio.imwrite('image.png', image)
     
 
-def path(action, body, hair, top, bottom):
+def run(action, body, hair, top, bottom):
 
     # body
     if body == "human": body = '0'
@@ -77,11 +77,11 @@ def path(action, body, hair, top, bottom):
     
     gif = display_image(name)
     
-    return 'image.png'
+    return 'image.png', 'image.png'
 
 
 gr.Interface(
-    path,
+    run,
     inputs=[
         gr.Radio(choices=["shoot", "slash", "spellcard", "thrust", "walk"], value="shoot"),
         gr.Radio(choices=["human", "alien"], value="human"),
@@ -89,6 +89,10 @@ gr.Interface(
         gr.Radio(choices=["brown", "blue", "white"], value="brown"),
         gr.Radio(choices=["white", "golden", "red", "silver"], value="white"),
     ],
-    outputs="image",
+    outputs=[
+        gr.components.Image(type="file", label="Avatar (Source)"),
+        gr.components.Image(type="file", label="Avatar (Target)")
+    ],
     live=False,
+    title="TransferVAE",
 ).launch()