Merge branch 'master' of https://github.com/hagabbar/vitamin_b

README.md

@@ -22,21 +22,30 @@ Note: This repository is a work in progress. No official release of code just ye
 
 ## Requirements
 
-Make sure to install basemap prior to installing all other packages.
+VItamin requires python3.6. You may use python3.6 by initializing a virtual environment.
 
-For model:
+```
+virtualenv -p python3.6 myenv
+source myenv/bin/activate
+pip install --upgrade pip
+```
 
-- tensorflow
-- tensorflow-probability
-- numpy
-- bilby
-- basemap
-- And other packages automatically installed via `pip install .`.
+Make sure to install basemap prior to installing all other packages.
 
 For installing basemap:
 - Install geos-3.3.3 from source
 - Once geos is installed, install basemap using `pip install git+https://github.com/matplotlib/basemap.git`
 
+For other required packages:
+```
+pip install -r requirements.txt
+```
+
+Install VItamin using pip:
+```
+pip install vitamin-b
+```
+
 ## Training
 
 To train an example model from the paper, try out the [demo](https://colab.research.google.com/github/hagabbar/OzGrav_demo/blob/master/OzGrav_VItamin_demo.ipynb).