From 8e5fb17063750ec39bd107c250a7c46a793e8890 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Mon, 14 Sep 2020 13:38:06 -0700
Subject: [PATCH] add badge

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f263d9c..7456aff 100644
--- a/README.md
+++ b/README.md
@@ -6,6 +6,8 @@ Implementation of <a href="https://arxiv.org/abs/2006.11239">Denoising Diffusion
 
 <img src="./sample.png" width="500px"><img>
 
+[![PyPI version](https://badge.fury.io/py/denoising-diffusion-pytorch.svg)](https://badge.fury.io/py/denoising-diffusion-pytorch)
+
 ## Install
 
 ```bash
@@ -73,8 +75,6 @@ trainer = Trainer(
 trainer.train()
 ```
 
-Todo: Command line tool for one-line training
-
 ## Citations
 
 ```bibtex