diff --git a/README.md b/README.md
index edfb6785..8ee1fd02 100644
--- a/README.md
+++ b/README.md
@@ -32,6 +32,7 @@ Below is a code snippet for doing so:
 
 ```python
 import model_search
+from model_search import constants
 from model_search import single_trainer
 from model_search.data import csv_data
 
@@ -41,7 +42,7 @@ trainer = single_trainer.SingleTrainer(
         logits_dimension=2,
         record_defaults=[0, 0, 0, 0],
         filename="model_search/data/testdata/csv_random_data.csv")),
-    spec=model_search.constants.DEFAULT_DNN)
+    spec=constants.DEFAULT_DNN)
 
 trainer.try_models(
     number_models=200,
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 00000000..e91b7310
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,8 @@
+six==1.15.0
+sklearn==0.0
+tensorflow==2.2.0
+absl-py==0.10.0
+tf-slim==1.1.0
+ml-metadata==0.26.0
+keras-tuner==1.0.2
+mock==4.0.3