Add Java and Kotlin API for punctuation models (#818)

This commit is contained in:
Fangjun Kuang
2024-04-26 22:06:48 +08:00
committed by GitHub
parent db25986240
commit 5407f880c0
19 changed files with 515 additions and 0 deletions

View File

@@ -0,0 +1,56 @@
// Copyright 2024 Xiaomi Corporation
package com.k2fsa.sherpa.onnx;
public class OfflinePunctuationModelConfig {
private final String ctTransformer;
private final int numThreads;
private final boolean debug;
private final String provider;
private OfflinePunctuationModelConfig(Builder builder) {
this.ctTransformer = builder.ctTransformer;
this.numThreads = builder.numThreads;
this.debug = builder.debug;
this.provider = builder.provider;
}
public static Builder builder() {
return new Builder();
}
public String getCtTransformer() {
return ctTransformer;
}
public static class Builder {
private String ctTransformer = "";
private int numThreads = 1;
private boolean debug = true;
private String provider = "cpu";
public OfflinePunctuationModelConfig build() {
return new OfflinePunctuationModelConfig(this);
}
public Builder setCtTransformer(String ctTransformer) {
this.ctTransformer = ctTransformer;
return this;
}
public Builder setNumThreads(int numThreads) {
this.numThreads = numThreads;
return this;
}
public Builder setDebug(boolean debug) {
this.debug = debug;
return this;
}
public Builder setProvider(String provider) {
this.provider = provider;
return this;
}
}
}