diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 7109bff7..e11c126d 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -36,6 +36,10 @@ jobs:
         run: |
           uv run pytest -v --cov=data_designer --cov-report=term-missing --cov-report=xml --cov-fail-under=90
 
+      - name: Update Coverage Badge
+        if: github.ref == format('refs/heads/{0}', github.event.repository.default_branch)
+        uses: we-cli/coverage-badge-action@v1.0.1
+
   lint:
     name: Lint and Format Check
     runs-on: ubuntu-latest
diff --git a/README.md b/README.md
index 8449e80b..dd4947b8 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,5 @@
 # 🎨 NeMo Data Designer
-[![CI](https://github.com/NVIDIA-NeMo/DataDesigner/actions/workflows/ci.yml/badge.svg)](https://github.com/NVIDIA-NeMo/DataDesigner/actions/workflows/ci.yml) [![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
+[![CI](https://github.com/NVIDIA-NeMo/DataDesigner/actions/workflows/ci.yml/badge.svg)](https://github.com/NVIDIA-NeMo/DataDesigner/actions/workflows/ci.yml) [![cov](https://NVIDIA-NeMo.github.io/DataDesigner/badges/coverage.svg)](https://github.com/NVIDIA-NeMo/DataDesigner/actions) [![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
 
 Create synthetic datasets from scratch.