diff --git a/test.ipynb b/test.ipynb
index 0039569..ca592e3 100644
--- a/test.ipynb
+++ b/test.ipynb
@@ -22,7 +22,6 @@
         "!pip install -q -r https://github.com/camenduru/stable-video-diffusion-colab/raw/main/requirements.txt\n",
         "!pip install -q -e generative-models\n",
         "!pip install -q -e git+https://github.com/Stability-AI/datapipelines@main#egg=sdata\n",
-        "!pip install gradio==3.50.2 -U\n",
         "\n",
         "!apt -y install -qq aria2\n",
         "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/vdo/stable-video-diffusion-img2vid-xt/resolve/main/svd_xt.safetensors?download=true -d /content/checkpoints -o svd_xt.safetensors\n",
@@ -103,8 +102,6 @@
         "from PIL import Image\n",
         "from torchvision.transforms import ToTensor\n",
         "from torchvision.transforms import functional as TF\n",
-        "\n",
-        "from sgm.inference.helpers import embed_watermark\n",
         "from sgm.util import default, instantiate_from_config\n",
         "\n",
         "def get_unique_embedder_keys_from_conditioner(conditioner):\n",
@@ -310,8 +307,6 @@
         "                    fps_id + 1,\n",
         "                    (samples.shape[-1], samples.shape[-2]),\n",
         "                )\n",
-        "\n",
-        "                samples = embed_watermark(samples)\n",
         "                vid = (\n",
         "                    (rearrange(samples, \"t c h w -> t h w c\") * 255)\n",
         "                    .cpu()\n",