fix broken imports.

PiperOrigin-RevId: 595666325 Change-Id: Ia043c31b33d2fe0818ad846ae0f59f4b57e64ede
SergeySandler · Feb 8, 2024 · 230a676 · 230a676
1 parent d8725ae
commit 230a676
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/colabs/tapir_demo.ipynb b/colabs/tapir_demo.ipynb
@@ -166,6 +166,25 @@
         "model_apply = jax.jit(model.apply)"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "Y1HkuvyT6SJF"
+      },
+      "outputs": [],
+      "source": [
+        "# @title Load an Exemplar Video {form-width: \"25%\"}\n",
+        "\n",
+        "%mkdir tapnet/examplar_videos\n",
+        "\n",
+        "!wget -P tapnet/examplar_videos http://storage.googleapis.com/dm-tapnet/horsejump-high.mp4\n",
+        "\n",
+        "video = media.read_video('tapnet/examplar_videos/horsejump-high.mp4')\n",
+        "height, width = video.shape[1:3]\n",
+        "media.show_video(video, fps=10)"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -213,25 +232,6 @@
         "  return points"
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "Y1HkuvyT6SJF"
-      },
-      "outputs": [],
-      "source": [
-        "# @title Load an Exemplar Video {form-width: \"25%\"}\n",
-        "\n",
-        "%mkdir tapnet/examplar_videos\n",
-        "\n",
-        "!wget -P tapnet/examplar_videos https://storage.googleapis.com/dm-tapnet/horsejump-high.mp4\n",
-        "\n",
-        "video = media.read_video('tapnet/examplar_videos/horsejump-high.mp4')\n",
-        "height, width = video.shape[1:3]\n",
-        "media.show_video(video, fps=10)"
-      ]
-    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -274,7 +274,7 @@
         "fig, ax = plt.subplots(figsize=(10, 5))\n",
         "ax.imshow(video[select_frame])\n",
         "ax.axis('off')\n",
-        "ax.set_title('You can select more than 1 points. After select enough points, run the next cell.')\n",
+        "ax.set_title('You can select more than 1 point. After selecting enough points, run the next cell.')\n",
         "\n",
         "select_points = []\n",
         "\n",