From bd6b33c325d3a8f9b4c5b706d071e133643db557 Mon Sep 17 00:00:00 2001
From: ahwar <20405797+Ahwar@users.noreply.github.com>
Date: Sat, 31 Aug 2024 16:56:59 +0500
Subject: [PATCH] ci: run inference with java

Use downloaded model artifact and
run Main.java with Gradle for inference
---
 .github/workflows/java-app.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/java-app.yml b/.github/workflows/java-app.yml
index ef9dc49..900018b 100644
--- a/.github/workflows/java-app.yml
+++ b/.github/workflows/java-app.yml
@@ -4,7 +4,7 @@ name: Java application
 
 on:
   push:
-    branches: ["main"]
+    branches: ["main", "g"]
   pull_request:
     branches: ["main"]
 
@@ -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