diff --git a/.github/workflows/pypi_release.yml b/.github/workflows/pypi_release.yml
new file mode 100644
index 000000000..1f976b2bf
--- /dev/null
+++ b/.github/workflows/pypi_release.yml
@@ -0,0 +1,31 @@
+# This workflow uses actions that are not certified by GitHub.
+# They are provided by a third-party and are governed by
+# separate terms of service, privacy policy, and support
+# documentation.
+
+name: Upload Python Package
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  deploy:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python
+        uses: actions/setup-python@v3
+        with:
+          python-version: "3.x"
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install build
+      - name: Build package
+        run: python -m build
+      - name: Publish package
+        uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
+        with:
+          user: __token__
+          password: ${{ secrets.PYPI_API_TOKEN }}
diff --git a/src/dalle_mini/__init__.py b/src/dalle_mini/__init__.py
index c25afaa99..a97a6406d 100644
--- a/src/dalle_mini/__init__.py
+++ b/src/dalle_mini/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "0.0.5"
+__version__ = "0.0.6"
 
 from .model import DalleBart, DalleBartProcessor