Update for lookahead in kaldi

alphacep · Nov 25, 2019 · 3934e8b · 3934e8b
1 parent 93ef15e
commit 3934e8b
Show file tree

Hide file tree

Showing 6 changed files with 66,407 additions and 66,390 deletions.
diff --git a/aars/kaldi-android-5.2.aar b/aars/kaldi-android-5.2.aar
diff --git a/app/src/main/java/org/kaldi/demo/KaldiActivity.java b/app/src/main/java/org/kaldi/demo/KaldiActivity.java
@@ -125,8 +125,9 @@ private static class RecognizeTask extends AsyncTask<Void, Void, String> {
         protected String doInBackground(Void... params) {
             KaldiRecognizer rec;
             long startTime = System.currentTimeMillis();
+            StringBuilder result = new StringBuilder();
             try {
-                rec = new KaldiRecognizer(activityReference.get().model);
+                rec = new KaldiRecognizer(activityReference.get().model, 16000.f);
 
                 InputStream ais = activityReference.get().getAssets().open("10001-90210-01803.wav");
                 if (ais.skip(44) != 44) {
@@ -135,12 +136,17 @@ protected String doInBackground(Void... params) {
                 byte[] b = new byte[4096];
                 int nbytes;
                 while ((nbytes = ais.read(b)) >= 0) {
-                    rec.AcceptWaveform(b, nbytes);
+                    if (rec.AcceptWaveform(b, nbytes)) {
+                        result.append(rec.Result());
+                    } else {
+                        result.append(rec.PartialResult());
+                    }
                 }
+                result.append(rec.FinalResult());
             } catch (IOException e) {
                 return "";
             }
-            return String.format(activityReference.get().getString(R.string.elapsed), rec.FinalResult(), (System.currentTimeMillis() - startTime));
+            return String.format(activityReference.get().getString(R.string.elapsed), result.toString(), (System.currentTimeMillis() - startTime));
         }
 
         @Override

diff --git a/models/src/main/assets/sync/model-android/Gr.fst b/models/src/main/assets/sync/model-android/Gr.fst
diff --git a/models/src/main/assets/sync/model-android/HCLr.fst b/models/src/main/assets/sync/model-android/HCLr.fst
diff --git a/models/src/main/assets/sync/model-android/disambig_tid.int b/models/src/main/assets/sync/model-android/disambig_tid.int
@@ -0,0 +1,11 @@
+12003
+12004
+12005
+12006
+12007
+12008
+12009
+12010
+12011
+12012
+12013