diff --git a/Dockerfile b/Dockerfile
index 6ac9876..d578f70 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ubuntu:18.04 as builder
+FROM ubuntu:20.04 as builder
 LABEL maintainer="Ravnoor Singh Gill <ravnoor@gmail.com>"
 
 RUN apt-get update && apt-get upgrade -y \
@@ -13,9 +13,9 @@ RUN apt-get update && apt-get upgrade -y \
 RUN mkdir /app
 WORKDIR /app
 
-RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-py37_23.1.0-1-Linux-x86_64.sh \
-    && /bin/bash Miniconda3-py37_23.1.0-1-Linux-x86_64.sh -b -p /opt/conda \
-    && rm Miniconda3-py37_23.1.0-1-Linux-x86_64.sh
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-py38_23.5.2-0-Linux-x86_64.sh \
+    && /bin/bash Miniconda3-py38_23.5.2-0-Linux-x86_64.sh -b -p /opt/conda \
+    && rm Miniconda3-py38_23.5.2-0-Linux-x86_64.sh
 
 ENV PATH=/opt/conda/bin:$PATH
 
@@ -28,7 +28,7 @@ RUN pip install -r /app/requirements.txt
 
 
 # production image
-FROM ubuntu:18.04
+FROM ubuntu:20.04
 
 ENV TZ=America/Montreal
 
diff --git a/README.md b/README.md
index 5ee67fa..b82a7a7 100644
--- a/README.md
+++ b/README.md
@@ -4,9 +4,9 @@
 
 <p align="center">
       <a href="https://www.python.org/">
-        <img src="https://img.shields.io/badge/Python-3.7-ff69b4.svg" /></a>
+        <img src="https://img.shields.io/badge/Python-3.7+-ff69b4.svg" /></a>
       <a href= "https://pytorch.org/">
-        <img src="https://img.shields.io/badge/PyTorch-1.7-2BAF2B.svg" /></a>
+        <img src="https://img.shields.io/badge/PyTorch-1.8%20LTS-2BAF2B.svg" /></a>
       <a href= "https://github.com/NOEL-MNI/deepMask/blob/main/LICENSE">
         <img src="https://img.shields.io/badge/License-BSD%203--Clause-blue.svg" /></a>
       <a href="https://doi.org/10.5281/zenodo.4521706">
@@ -41,11 +41,11 @@ Milletari, F., Navab, N., & Ahmadi, S. A. (2016, October). [V-net: Fully convolu
 ```
 
 ## Pre-requisites
-###TODO: Update version requirements
 ```console
-1. Python >= 3.5
-2. Pytorch >= 1.0.0
-3. Nibabel >= 1.14.0
+1. Python >= 3.7
+2. PyTorch (LTS) <= 1.8.2
+3. ANTsPy
+4. ANTsPyNet
 ```
 
 ## Installation
diff --git a/app/requirements.txt b/app/requirements.txt
index ca79cb8..7461e28 100644
--- a/app/requirements.txt
+++ b/app/requirements.txt
@@ -11,6 +11,7 @@ scikit-image==0.19.2
 scikit-learn==1.0.2
 scipy==1.7.3
 setproctitle==1.2.3
+tensorflow_probability==0.19.0
 torchsummary==1.5.1
 torch==1.8.2
 torchvision==0.9.2
\ No newline at end of file