diff --git a/Brain_Tumour_Classification.ipynb b/Brain_Tumour_Classification.ipynb
index ed6f914..406358f 100644
--- a/Brain_Tumour_Classification.ipynb
+++ b/Brain_Tumour_Classification.ipynb
@@ -919,7 +919,7 @@
         "    # 16 x 16 x 32\n",
         "    # flatten\n",
         "    nn.Flatten(),\n",
-        "    # 1 x 16384\n",
+        "    # 1 x 8192\n",
         "    nn.Linear(16*16*32, 512),\n",
         "    nn.ReLU(),\n",
         "    nn.Dropout(p=0.7),\n",
@@ -932,7 +932,7 @@
       "metadata": {
         "id": "G44l3Bijgl0H"
       },
-      "execution_count": 527,
+      "execution_count": 565,
       "outputs": []
     },
     {
@@ -945,9 +945,9 @@
           "base_uri": "https://localhost:8080/"
         },
         "id": "PB-qy2909Faq",
-        "outputId": "1c18c4c7-4c41-40a1-b4d7-925ba3337732"
+        "outputId": "01b7024a-b8a1-436d-e371-33c260a083c3"
       },
-      "execution_count": 528,
+      "execution_count": 566,
       "outputs": [
         {
           "output_type": "stream",
@@ -1501,7 +1501,7 @@
       "collapsed_sections": [],
       "name": "Brain Tumour Classification",
       "provenance": [],
-      "authorship_tag": "ABX9TyMVaqV7S/yULeuAfxMmvKfL",
+      "authorship_tag": "ABX9TyOL/hIyPIitfBep4jbFIvHw",
       "include_colab_link": true
     },
     "kernelspec": {