diff --git a/examples/mnist-pytorch/API_Example.ipynb b/examples/mnist-pytorch/API_Example.ipynb
index fabcb8c1a..2be720c81 100644
--- a/examples/mnist-pytorch/API_Example.ipynb
+++ b/examples/mnist-pytorch/API_Example.ipynb
@@ -14,7 +14,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 50,
+   "execution_count": 1,
    "id": "743dfe47",
    "metadata": {},
    "outputs": [],
@@ -38,7 +38,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 51,
+   "execution_count": 2,
    "id": "1061722d",
    "metadata": {},
    "outputs": [],
@@ -58,7 +58,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 52,
+   "execution_count": 3,
    "id": "5107f6f9",
    "metadata": {},
    "outputs": [
@@ -66,7 +66,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "{'committed_at': 'Wed, 27 Mar 2024 22:00:56 GMT', 'id': '66049718d5d8d370bf266899', 'key': 'models', 'model': '8762bb36-ffcc-4591-ad62-2758c6284a5d', 'parent_model': None, 'session_id': None}\n"
+      "{'committed_at': 'Mon, 08 Apr 2024 13:05:03 GMT', 'id': '6613eb7ffe38d7586cf7c3e5', 'key': 'models', 'model': 'dab2722b-31c3-4994-a12c-4bbb2f8f29aa', 'parent_model': None, 'session_id': None}\n"
      ]
     }
    ],