diff --git a/requirements.txt b/requirements.txt
index 3c9d012..62ad006 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,6 @@
-tensorflow
+pandas
+scikit-learn
+autopep8
+tqdm
+jinja2>=2.8
 pyyaml
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 509d592..d9cd574 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
 setup(
     name='automl_gs',
     packages=['automl_gs'],  # this must be the same as the name above
-    version='0.2',
+    version='0.2.1',
     description='Provide an input CSV and a target field to predict, ' \
     'generate a model + code to run it.',
     long_description=long_description,
@@ -39,5 +39,5 @@
     },
     python_requires='>=3.5',
     include_package_data=True,
-    install_requires=['pandas', 'scikit-learn', 'autopep8', 'tqdm', 'jinja2>=2.8', 'tensorflow', 'pyyaml']
+    install_requires=['pandas', 'scikit-learn', 'autopep8', 'tqdm', 'jinja2>=2.8', 'pyyaml']
 )