Update app.py

app.py

@@ -257,7 +257,7 @@ def MyPlot(frame_id, src_orig, tar_orig, src_recon, tar_recon, src_Zt, tar_Zt, s
     axs[1, 3].imshow(src_Zf_tar_Zt)
     axs[1, 3].axis('off')
     
-    plt.subplots_adjust(hspace=0.06, wspace=0.05)
+    plt.subplots_adjust(hspace=0.0, wspace=0.0)
     
     save_name = 'MyPlot_{}.png'.format(frame_id)
     
@@ -382,7 +382,7 @@ def run(source, action_source, hair_source, top_source, bottom_source, target, a
         
     a = concat('MyPlot_')
     
-    return 'demo.gif', None
+    return 'demo.gif'
 
 
 gr.Interface(
@@ -409,13 +409,6 @@ gr.Interface(
     ],
     outputs=[
         gr.components.Image(type="file", label="Domain Disentanglement"),
-        gr.Markdown(
-            """
-            Row 1 - Original source (left) and target (right) inputs\n
-            Row 2 - Reconstruction outputs\n
-            Row 3 -
-            """
-        ),
     ],
     live=True,
     cache_examples=True,