Update app.py

app.py

@@ -90,8 +90,8 @@ gr.Interface(
         gr.Radio(choices=["white", "golden", "red", "silver"], value="white"),
     ],
     outputs=[
-        gr.components.Image(type="file", shape=[16, 16], label="Avatar (Source)"),
-        gr.components.Image(type="file", shape=[16, 16], label="Avatar (Target)")
+        gr.components.Image(type="file", label="Avatar (Source)"),
+        gr.components.Gallery(['image.png', 'image.png'], label="Avatar (Target)")
     ],
     live=True,
     title="TransferVAE",