From f8873ab2e03b3f23c24b2e99b108ed53880b5349 Mon Sep 17 00:00:00 2001
From: Alexander Pivovarov <pivovaa@amazon.com>
Date: Tue, 25 Jun 2019 12:11:24 -0700
Subject: [PATCH] Update tflite wheel version to 1.13.1

---
 docker/install/ubuntu_install_tflite.sh |  2 +-
 tutorials/frontend/from_tflite.py       | 28 ++++++++++++++++++-------
 2 files changed, 21 insertions(+), 9 deletions(-)

diff --git a/docker/install/ubuntu_install_tflite.sh b/docker/install/ubuntu_install_tflite.sh
index 802fb3b87d8c0..df65753aace4c 100755
--- a/docker/install/ubuntu_install_tflite.sh
+++ b/docker/install/ubuntu_install_tflite.sh
@@ -43,7 +43,7 @@ import setuptools
 
 setuptools.setup(
     name="tflite",
-    version="0.0.1",
+    version="1.13.1",
     author="google",
     author_email="google@google.com",
     description="TFLite",
diff --git a/tutorials/frontend/from_tflite.py b/tutorials/frontend/from_tflite.py
index 5a8525133f7c8..8348f35ab1bbb 100644
--- a/tutorials/frontend/from_tflite.py
+++ b/tutorials/frontend/from_tflite.py
@@ -29,20 +29,32 @@
 
     pip install flatbuffers --user
 
+
 To install TFlite packages, you could use our prebuilt wheel:
 
 .. code-block:: bash
 
     # For python3:
-    wget https://raw.githubusercontent.com/dmlc/web-data/master/tensorflow/tflite/whl/tflite-0.0.1-py3-none-any.whl
-    pip install tflite-0.0.1-py3-none-any.whl --user
+    wget https://raw.githubusercontent.com/dmlc/web-data/master/tensorflow/tflite/whl/tflite-1.13.1-py3-none-any.whl
+    pip3 install tflite-1.13.1-py3-none-any.whl --user
 
     # For python2:
-    wget https://raw.githubusercontent.com/dmlc/web-data/master/tensorflow/tflite/whl/tflite-0.0.1-py2-none-any.whl
-    pip install tflite-0.0.1-py2-none-any.whl --user
+    wget https://raw.githubusercontent.com/dmlc/web-data/master/tensorflow/tflite/whl/tflite-1.13.1-py2-none-any.whl
+    pip install tflite-1.13.1-py2-none-any.whl --user
+
+
+or you could clone FrozenGene/tflite repo which contains generated tflite python module
+
+.. code-block:: bash
+
+    # clone FrozenGene/tflite repo to tflite_repo folder
+    git clone https://github.com/FrozenGene/tflite.git tflite_repo
+
+    # Add git repo folder (which contains generated tflite module) to PYTHONPATH.
+    export PYTHONPATH=${PYTHONPATH:+$PYTHONPATH:}$(pwd)/tflite_repo
 
 
-or you could generate TFLite package by yourself. The steps are as following:
+or you could generate TFLite package yourself. The steps are the following:
 
 .. code-block:: bash
 
@@ -57,13 +69,13 @@
     # Generate TFLite package.
     flatc --python schema.fbs
 
-    # Add it to PYTHONPATH.
-    export PYTHONPATH=/path/to/tflite
+    # Add current folder (which contains generated tflite module) to PYTHONPATH.
+    export PYTHONPATH=${PYTHONPATH:+$PYTHONPATH:}$(pwd)
 
 
 Now please check if TFLite package is installed successfully, ``python -c "import tflite"``
 
-Below you can find an example for how to compile TFLite model using TVM.
+Below you can find an example on how to compile TFLite model using TVM.
 """
 ######################################################################
 # Utils for downloading and extracting zip files