From 04b5b113264f91399a1775533fd49d80167395b1 Mon Sep 17 00:00:00 2001
From: Aleksey Morozov <36787333+amrzv@users.noreply.github.com>
Date: Sun, 11 Jul 2021 12:46:55 +0300
Subject: [PATCH] showing original and processed video inline

---
 SimSwap colab.ipynb | 41 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 40 insertions(+), 1 deletion(-)

diff --git a/SimSwap colab.ipynb b/SimSwap colab.ipynb
index 850989b..bf74b2e 100644
--- a/SimSwap colab.ipynb	
+++ b/SimSwap colab.ipynb	
@@ -351,6 +351,25 @@
         }
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "from IPython.display import HTML\n",
+        "from base64 import b64encode\n",
+        "\n",
+        "def show_mp4(filename, width=720):\n",
+        "    mp4 = open(filename, 'rb').read()\n",
+        "    data_url = \"data:video/mp4;base64,\" + b64encode(mp4).decode()\n",
+        "    display(HTML(\"\"\"\n",
+        "    <video width=\"%d\" controls autoplay loop>\n",
+        "        <source src=\"%s\" type=\"video/mp4\">\n",
+        "    </video>\n",
+        "    \"\"\" % (width, data_url)))   "
+      ]
+    },
     {
       "cell_type": "code",
       "metadata": {
@@ -564,12 +583,32 @@
         }
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "show_mp4(opt.video_path)  # original video"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "img_a_align_crop_pil  # swappable face"
+      ]
+    },
     {
       "cell_type": "code",
       "metadata": {
         "id": "Rty2GsyZZrI6"
       },
-      "source": [],
+      "source": [
+        "show_mp4(opt.output_path)  # processed video"
+      ],
       "execution_count": null,
       "outputs": []
     }