diff --git a/demo.ipynb b/demo.ipynb
index 422dddf..06b70fb 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -76,6 +76,15 @@
     "!pip install keras==2.2.4"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%tensorflow_version 1.x"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
@@ -1115,20 +1124,6 @@
     "print(\"Gender: female\") if gender == 0 else print(\"Gender: male\")\n",
     "print(f\"Age: {str(age)}\")"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {
@@ -1158,4 +1153,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
+}
\ No newline at end of file