From 417f6a8033c860cf8dc5c09ffd1de776a4080d4d Mon Sep 17 00:00:00 2001
From: Aditya Nikhil <aditya.digala@gmail.com>
Date: Sun, 31 Jan 2021 23:44:17 +0530
Subject: [PATCH] Update tfhub_models.ipynb

---
 compare_gan/src/tfhub_models.ipynb | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/compare_gan/src/tfhub_models.ipynb b/compare_gan/src/tfhub_models.ipynb
index 4c562e1..f73cfb0 100644
--- a/compare_gan/src/tfhub_models.ipynb
+++ b/compare_gan/src/tfhub_models.ipynb
@@ -121,9 +121,10 @@
         "import numpy as np\n",
         "import pandas as pd\n",
         "\n",
-        "import tensorflow as tf\n",
+        "import tensorflow.compat.v1 as tf\n",
         "import tensorflow_hub as hub\n",
         "\n",
+        "tf.disable_v2_behavior()\n",
         "\n",
         "tf.logging.set_verbosity(tf.logging.ERROR)\n",
         "\n",
@@ -339,4 +340,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}