Add Kotlin and Java API for online punctuation models (#1936)
This commit is contained in:
41
java-api-examples/OnlineAddPunctuation.java
Normal file
41
java-api-examples/OnlineAddPunctuation.java
Normal file
@@ -0,0 +1,41 @@
|
||||
// Copyright 2025 Xiaomi Corporation
|
||||
|
||||
// This file shows how to use a punctuation model to add punctuations to text.
|
||||
//
|
||||
// The model supports ONLY English.
|
||||
import com.k2fsa.sherpa.onnx.*;
|
||||
|
||||
public class OnlineAddPunctuation {
|
||||
public static void main(String[] args) {
|
||||
// please download the model from
|
||||
// https://github.com/k2-fsa/sherpa-onnx/releases/download/punctuation-models/sherpa-onnx-online-punct-en-2024-08-06.tar.bz2
|
||||
String model = "./sherpa-onnx-online-punct-en-2024-08-06/model.int8.onnx";
|
||||
String bpeVocab = "./sherpa-onnx-online-punct-en-2024-08-06/bpe.vocab";
|
||||
OnlinePunctuationModelConfig modelConfig =
|
||||
OnlinePunctuationModelConfig.builder()
|
||||
.setCnnBilstm(model)
|
||||
.setBpeVocab(bpeVocab)
|
||||
.setNumThreads(1)
|
||||
.setDebug(true)
|
||||
.build();
|
||||
OnlinePunctuationConfig config =
|
||||
OnlinePunctuationConfig.builder().setModel(modelConfig).build();
|
||||
|
||||
OnlinePunctuation punct = new OnlinePunctuation(config);
|
||||
|
||||
String[] sentences =
|
||||
new String[] {
|
||||
"how are you doing fantastic thank you how about you",
|
||||
"The African blogosphere is rapidly expanding bringing more voices online in the form of"
|
||||
+ " commentaries opinions analyses rants and poetry",
|
||||
};
|
||||
|
||||
System.out.println("---");
|
||||
for (String text : sentences) {
|
||||
String out = punct.addPunctuation(text);
|
||||
System.out.printf("Input: %s\n", text);
|
||||
System.out.printf("Output: %s\n", out);
|
||||
System.out.println("---");
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user