mirror of
https://github.com/k2-fsa/sherpa-onnx.git
synced 2026-11-11 05:20:52 +08:00
39 lines
1.5 KiB
Java
39 lines
1.5 KiB
Java
// Copyright 2026 Matias Lin
|
|
|
|
// This file shows how to use a CATT diacritization model to add diacritics to Arabic text.
|
|
import com.k2fsa.sherpa.onnx.*;
|
|
|
|
public class OfflineAddDiacritics {
|
|
public static void main(String[] args) {
|
|
// please download the model from
|
|
// https://github.com/abjadai/catt/releases/download/v2/eo_model_onnx.zip
|
|
String cattEncoder = "./catt_eo_model_onnx/encoder.onnx";
|
|
String cattDecoder = "./catt_eo_model_onnx/decoder.onnx";
|
|
OfflineDiacritizationModelConfig modelConfig =
|
|
OfflineDiacritizationModelConfig.builder()
|
|
.setCattEncoder(cattEncoder)
|
|
.setCattDecoder(cattDecoder)
|
|
.setNumThreads(1)
|
|
.setDebug(true)
|
|
.build();
|
|
OfflineDiacritizationConfig config =
|
|
OfflineDiacritizationConfig.builder().setModel(modelConfig).build();
|
|
|
|
OfflineDiacritization diacrt = new OfflineDiacritization(config);
|
|
|
|
String[] sentences =
|
|
new String[] {
|
|
"وقالت مجلة نيوزويك الأمريكية التحديث الجديد ل إنستجرام يمكن أن يساهم في إيقاف وكشف الحسابات المزورة بسهولة شديدة",
|
|
"اللغة العربية من أقدم اللغات السامية",
|
|
};
|
|
|
|
System.out.println("---");
|
|
for (String text : sentences) {
|
|
String out = diacrt.addDiacritics(text);
|
|
System.out.printf("Input: %s\n", text);
|
|
System.out.printf("Output: %s\n", out);
|
|
System.out.println("---");
|
|
}
|
|
}
|
|
}
|