Few changes on notebook

flexnlp/notebooks/Federated IMDb PT using FLExible with a GRU.ipynb

@@ -665,7 +665,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Run the following cell in order to train the model for multiple rounds, selecting ya"
+    "Run the following cell in order to train the model for multiple rounds, selecting the rounds you want to train."
    ]
   },
   {
@@ -676,6 +676,15 @@
    "source": [
     "train_n_rounds(n_rounds=5)"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# END\n",
+    "\n",
+    "In this notebook we have shown how to train a BiGRU model using the FLEX library. As an example, the BiGRU is trained in a few rounds, but we recommend to try out different settings."
+   ]
   }
  ],
  "metadata": {