Skip to content

Commit

Permalink
ci: run inference with java
Browse files Browse the repository at this point in the history
Use downloaded model artifact and
run Main.java with Gradle for inference
  • Loading branch information
Ahwar committed Aug 31, 2024
1 parent db574ce commit bd6b33c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/java-app.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ name: Java application

on:
push:
branches: ["main"]
branches: ["main", "g"]
pull_request:
branches: ["main"]

Expand All @@ -28,7 +28,7 @@ jobs:
- name: Hugging Face Model to ONNX
run: |
# Convert Hugging Face Model to ONNX
python -m transformers.onnx --opset 16 --atol 0.005 --feature=token-classification --model=xlm-roberta-large-finetuned-conll03-english onnx_model/
python -m transformers.onnx --opset 16 --atol 0.005 --feature=token-classification --model=vitus9988/klue-roberta-base-ner-identified onnx_model/
# Upload the ONNX model as an artifact
- name: Upload ONNX model
Expand Down

0 comments on commit bd6b33c

Please sign in to comment.