Update generate_conda_file.py

tools/generate_conda_file.py

@@ -87,13 +87,13 @@ PIP_BASE = {
     "pymanopt": "pymanopt==0.2.5",
     "xlearn": "xlearn==0.40a1",
     "transformers": "transformers==2.5.0",
-    "tensorflow": "tensorflow==1.15.2",
+    "tensorflow": "tensorflow>=2.2.2,<2.4",
     "scrapbook": "scrapbook>=0.5.0",
 }
 
 PIP_GPU = {
     "nvidia-ml-py3": "nvidia-ml-py3>=7.352.0",
-    "tensorflow": "tensorflow-gpu==1.15.2",
+    "tensorflow": "tensorflow-gpu>=2.2.2,<2.4",
 }
 PIP_PYSPARK = {"databricks-cli": "databricks-cli==0.8.6"}