Skip to content

Commit

Permalink
fix(action): support multiple models (#64)
Browse files Browse the repository at this point in the history
* fix(action): support multiple models

* update outdir
  • Loading branch information
swpfY authored Sep 6, 2024
1 parent a9cd1b5 commit 54913d3
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions .github/workflows/build-image.yml
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,10 @@ name: build image and push
on:
push:
branches:
- action
- master
tags:
- '*'

env:
MODEL_URL: https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/download/pretrained-model/hivision_modnet.onnx

jobs:
docker:
runs-on: ubuntu-latest
Expand All @@ -18,8 +15,8 @@ jobs:
- name: Checkout
uses: actions/checkout@v4

- name: Download model
run: wget -O hivision_modnet.onnx ${{ vars.MODEL_URL || env.MODEL_URL }}
- name: Download models
run: echo ${{ vars.MODEL_URLS }} | xargs -n 1 -I {} wget -P hivision/creator/weights {}

- name: Set up QEMU
uses: docker/setup-qemu-action@v3
Expand Down

0 comments on commit 54913d3

Please sign in to comment.