Update readme.md

readme.md

@@ -1,9 +1,10 @@
-# DFlat-pytorch
+# DFlat-pytorch 
 ## An End-to-End Design Framework for Diffractive Optics and Metasurface-Based Vision Systems
 ![Python](https://img.shields.io/badge/python-3670A0?style=for-the-badge&logo=python&logoColor=ffdd54)
 ![PyTorch](https://img.shields.io/badge/PyTorch-%23EE4C2C.svg?style=for-the-badge&logo=PyTorch&logoColor=white)
 [![License: GPL v3](https://img.shields.io/badge/License-GPLv3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0)
 [![Pytest Workflow](https://github.com/DeanHazineh/DFlat-pytorch/actions/workflows/pytest.yml/badge.svg?branch=dflat_v2.0.0)](https://github.com/DeanHazineh/DFlat-pytorch/actions/workflows/pytest.yml)
+![Version](https://img.shields.io/badge/version-2.0.0-blue)
 
 - This repository is the home for the new and maintained version of DFlat. It replaces DFlat-Tensorflow.
 - Note that this package is no longer a direct port of pytorch-tensorflow but is a complete rewrite released in early 2024.