diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index c8da614..e82b162 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -56,6 +56,8 @@ jobs:
           labels: ${{ steps.meta.outputs.labels }}
           cache-from: type=gha
           cache-to: type=gha,mode=max
+          build-args: |
+            VERSION=${{ github.ref_name }}
 
       - name: Generate artifact attestation
         if: github.event_name != 'pull_request' && steps.build.outputs.digest != ''
diff --git a/Dockerfile b/Dockerfile
index 568a645..8b93dd9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,13 +1,15 @@
 FROM node:20-alpine AS builder
 
+ARG VERSION=1.0.0
 WORKDIR /build
 COPY ./web .
 RUN npm install
-RUN npm run build
+RUN VITE_VERSION=${VERSION} npm run build
 
 
 FROM golang:alpine AS builder2
 
+ARG VERSION=1.0.0
 ENV GO111MODULE=on \
     CGO_ENABLED=0 \
     GOOS=linux
@@ -19,7 +21,7 @@ RUN go mod download
 
 COPY . .
 COPY --from=builder /build/dist ./web/dist
-RUN go build -ldflags "-s -w " -o gpt-load
+RUN go build -ldflags "-s -w -X gpt-load/internal/version.Version=${VERSION}" -o gpt-load
 
 
 FROM alpine
diff --git a/internal/version/version.go b/internal/version/version.go
new file mode 100644
index 0000000..8773a74
--- /dev/null
+++ b/internal/version/version.go
@@ -0,0 +1,3 @@
+package version
+
+var Version = "1.0.0"