From b6e1c4dc80441c56f6cb293f1e2d58317356e6ee Mon Sep 17 00:00:00 2001
From: Yury Kozyrev <urakozz@me.com>
Date: Mon, 25 Mar 2019 21:18:35 +0100
Subject: [PATCH] Update index.js

---
 src/index.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/index.js b/src/index.js
index c494316..269fb95 100644
--- a/src/index.js
+++ b/src/index.js
@@ -22,7 +22,7 @@ const DEFAULT_MODEL_LOCATION =
  * @param {String} url Tiny YOLO Model URL
  */
 export async function downloadModel(url = DEFAULT_MODEL_LOCATION) {
-  return await tf.loadModel(url);
+  return await tf.loadLayersModel(url);
 }
 
 /**