diff --git a/CaptionFL.ipynb b/CaptionFL.ipynb
index 60f252b..9eac3b0 100644
--- a/CaptionFL.ipynb
+++ b/CaptionFL.ipynb
@@ -111,8 +111,18 @@
   "language_info": {
    "name": "python"
   },
-  "orig_nbformat": 4
+  "orig_nbformat": 4,
+  "colab": {
+   "provenance": [],
+   "machine_shape": "hm",
+   "gpuType": "T4"
+ },
+  "kernelspec": {
+   "name": "python3",
+   "display_name": "Python 3"
+  },
+  "accelerator": "GPU"
  },
  "nbformat": 4,
- "nbformat_minor": 2
-}
+ "nbformat_minor": 0
+}
\ No newline at end of file