diff --git a/Example_prediction.ipynb b/Example_prediction.ipynb
index b02bc2e..ebe6313 100644
--- a/Example_prediction.ipynb
+++ b/Example_prediction.ipynb
@@ -24,7 +24,7 @@
     "\n",
     "from utils.utils import get_yolo_boxes, makedirs\n",
     "from utils.bbox import draw_boxes\n",
-    "from keras.models import load_model\n",
+    "from tensorflow.keras.models import load_model\n",
     "from tqdm import tqdm\n",
     "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
@@ -302,7 +302,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.8"
+   "version": "3.7.7"
   }
  },
  "nbformat": 4,