diff --git a/.github/workflows/build-image.yml b/.github/workflows/build-image.yml
index 739e6096..a3ef331d 100644
--- a/.github/workflows/build-image.yml
+++ b/.github/workflows/build-image.yml
@@ -14,7 +14,7 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Download models
-        run: echo ${{ vars.MODEL_URLS }} | tr ' ' '\n' | xargs -n 1 -I {} wget -P "hivision/creator/weights" {}
+        run: python scripts/download_model.py
 
       - name: Set up QEMU
         uses: docker/setup-qemu-action@v3