diff --git a/15_processing_sequences_using_rnns_and_cnns.ipynb b/15_processing_sequences_using_rnns_and_cnns.ipynb
index 8441ada..b846b42 100644
--- a/15_processing_sequences_using_rnns_and_cnns.ipynb
+++ b/15_processing_sequences_using_rnns_and_cnns.ipynb
@@ -1857,6 +1857,13 @@
     "Now let's write a function that will generate a new chorale. We will give it a few seed chords, it will convert them to arpegios (the format expected by the model), and use the model to predict the next note, then the next, and so on. In the end, it will group the notes 4 by 4 to create chords again, and return the resulting chorale."
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "**Warning**: `model.predict_classes(X)` is deprecated. It is replaced with `np.argmax(model.predict(X), axis=-1)`."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 94,
diff --git a/17_autoencoders_and_gans.ipynb b/17_autoencoders_and_gans.ipynb
index ea4fe9f..af967c2 100644
--- a/17_autoencoders_and_gans.ipynb
+++ b/17_autoencoders_and_gans.ipynb
@@ -1633,7 +1633,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 144,
+   "execution_count": 77,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1644,7 +1644,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 203,
+   "execution_count": 78,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1662,7 +1662,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 204,
+   "execution_count": 79,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1671,7 +1671,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 205,
+   "execution_count": 80,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1685,7 +1685,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 206,
+   "execution_count": 81,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1694,7 +1694,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 207,
+   "execution_count": 82,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1707,7 +1707,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 208,
+   "execution_count": 83,
    "metadata": {},
    "outputs": [],
    "source": [