Fangjun Kuang
Committed by GitHub

Fix Java API examples (#883)

@@ -13,6 +13,7 @@ java_files += FeatureConfig.java @@ -13,6 +13,7 @@ java_files += FeatureConfig.java
13 java_files += OnlineLMConfig.java 13 java_files += OnlineLMConfig.java
14 java_files += OnlineParaformerModelConfig.java 14 java_files += OnlineParaformerModelConfig.java
15 java_files += OnlineZipformer2CtcModelConfig.java 15 java_files += OnlineZipformer2CtcModelConfig.java
  16 +java_files += OnlineNeMoCtcModelConfig.java
16 java_files += OnlineTransducerModelConfig.java 17 java_files += OnlineTransducerModelConfig.java
17 java_files += OnlineModelConfig.java 18 java_files += OnlineModelConfig.java
18 java_files += OnlineCtcFstDecoderConfig.java 19 java_files += OnlineCtcFstDecoderConfig.java
@@ -7,6 +7,7 @@ public class OnlineModelConfig { @@ -7,6 +7,7 @@ public class OnlineModelConfig {
7 private final OnlineTransducerModelConfig transducer; 7 private final OnlineTransducerModelConfig transducer;
8 private final OnlineParaformerModelConfig paraformer; 8 private final OnlineParaformerModelConfig paraformer;
9 private final OnlineZipformer2CtcModelConfig zipformer2Ctc; 9 private final OnlineZipformer2CtcModelConfig zipformer2Ctc;
  10 + private final OnlineNeMoCtcModelConfig neMoCtc;
10 private final String tokens; 11 private final String tokens;
11 private final int numThreads; 12 private final int numThreads;
12 private final boolean debug; 13 private final boolean debug;
@@ -17,6 +18,7 @@ public class OnlineModelConfig { @@ -17,6 +18,7 @@ public class OnlineModelConfig {
17 this.transducer = builder.transducer; 18 this.transducer = builder.transducer;
18 this.paraformer = builder.paraformer; 19 this.paraformer = builder.paraformer;
19 this.zipformer2Ctc = builder.zipformer2Ctc; 20 this.zipformer2Ctc = builder.zipformer2Ctc;
  21 + this.neMoCtc = builder.neMoCtc;
20 this.tokens = builder.tokens; 22 this.tokens = builder.tokens;
21 this.numThreads = builder.numThreads; 23 this.numThreads = builder.numThreads;
22 this.debug = builder.debug; 24 this.debug = builder.debug;
@@ -40,6 +42,10 @@ public class OnlineModelConfig { @@ -40,6 +42,10 @@ public class OnlineModelConfig {
40 return zipformer2Ctc; 42 return zipformer2Ctc;
41 } 43 }
42 44
  45 + public OnlineNeMoCtcModelConfig getNeMoCtc() {
  46 + return neMoCtc;
  47 + }
  48 +
43 public String getTokens() { 49 public String getTokens() {
44 return tokens; 50 return tokens;
45 } 51 }
@@ -64,6 +70,7 @@ public class OnlineModelConfig { @@ -64,6 +70,7 @@ public class OnlineModelConfig {
64 private OnlineParaformerModelConfig paraformer = OnlineParaformerModelConfig.builder().build(); 70 private OnlineParaformerModelConfig paraformer = OnlineParaformerModelConfig.builder().build();
65 private OnlineTransducerModelConfig transducer = OnlineTransducerModelConfig.builder().build(); 71 private OnlineTransducerModelConfig transducer = OnlineTransducerModelConfig.builder().build();
66 private OnlineZipformer2CtcModelConfig zipformer2Ctc = OnlineZipformer2CtcModelConfig.builder().build(); 72 private OnlineZipformer2CtcModelConfig zipformer2Ctc = OnlineZipformer2CtcModelConfig.builder().build();
  73 + private OnlineNeMoCtcModelConfig neMoCtc = OnlineNeMoCtcModelConfig.builder().build();
67 private String tokens = ""; 74 private String tokens = "";
68 private int numThreads = 1; 75 private int numThreads = 1;
69 private boolean debug = true; 76 private boolean debug = true;
@@ -89,6 +96,11 @@ public class OnlineModelConfig { @@ -89,6 +96,11 @@ public class OnlineModelConfig {
89 return this; 96 return this;
90 } 97 }
91 98
  99 + public Builder setNeMoCtc(OnlineNeMoCtcModelConfig neMoCtc) {
  100 + this.neMoCtc = neMoCtc;
  101 + return this;
  102 + }
  103 +
92 public Builder setTokens(String tokens) { 104 public Builder setTokens(String tokens) {
93 this.tokens = tokens; 105 this.tokens = tokens;
94 return this; 106 return this;
  1 +// Copyright 2024 Xiaomi Corporation
  2 +package com.k2fsa.sherpa.onnx;
  3 +
  4 +public class OnlineNeMoCtcModelConfig {
  5 + private final String model;
  6 +
  7 + private OnlineNeMoCtcModelConfig(Builder builder) {
  8 + this.model = builder.model;
  9 + }
  10 +
  11 + public static Builder builder() {
  12 + return new Builder();
  13 + }
  14 +
  15 + public String getModel() {
  16 + return model;
  17 + }
  18 +
  19 + public static class Builder {
  20 + private String model = "";
  21 +
  22 + public OnlineNeMoCtcModelConfig build() {
  23 + return new OnlineNeMoCtcModelConfig(this);
  24 + }
  25 +
  26 + public Builder setModel(String model) {
  27 + this.model = model;
  28 + return this;
  29 + }
  30 + }
  31 +
  32 +}