From 29cd6d5b902ebdfece2f5d546d264b0b6818d705 Mon Sep 17 00:00:00 2001
From: jangmin-deepx <67268979+jangmin-deepx@users.noreply.github.com>
Date: Wed, 23 Jun 2021 17:28:14 +0900
Subject: [PATCH] Fix Missing kwargs in prepare method

tensor_dict will be empty if gen_tensor_dict kwarg is missing
---
 tutorials/OnnxTensorflowImport.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tutorials/OnnxTensorflowImport.ipynb b/tutorials/OnnxTensorflowImport.ipynb
index 433ce2a..e066d7c 100644
--- a/tutorials/OnnxTensorflowImport.ipynb
+++ b/tutorials/OnnxTensorflowImport.ipynb
@@ -39,7 +39,7 @@
     "\n",
     "warnings.filterwarnings('ignore') # Ignore all the warning messages in this tutorial\n",
     "model = onnx.load('assets/super_resolution.onnx') # Load the ONNX file\n",
-    "tf_rep = prepare(model) # Import the ONNX model to Tensorflow"
+    "tf_rep = prepare(model, gen_tensor_dict=True) # Import the ONNX model to Tensorflow"
    ]
   },
   {