diff --git a/runtime/device/android/wenet/README.md b/runtime/device/android/wenet/README.md index 2485dd2eb..7c62d1e72 100644 --- a/runtime/device/android/wenet/README.md +++ b/runtime/device/android/wenet/README.md @@ -24,7 +24,7 @@ Or you can train your own model using WeNet training pipeline on your data. ### 2) Build APK -When your model is ready, put `final.zip` and `words.txt` into Android assets (`app/src/main/assets`) folder, +When your model is ready, put `final.zip` and `units.txt` into Android assets (`app/src/main/assets`) folder, then just build and run the APK. Here is a gif demo, which shows how our on-device streaming e2e ASR runs with low latency. Please note the wifi and data has been disabled in the demo so there is no network connection ^\_^. diff --git a/runtime/device/android/wenet/app/src/main/assets/README.md b/runtime/device/android/wenet/app/src/main/assets/README.md index 6645c402d..2d78237d0 100644 --- a/runtime/device/android/wenet/app/src/main/assets/README.md +++ b/runtime/device/android/wenet/app/src/main/assets/README.md @@ -1 +1 @@ -put final.zip and words.txt here. +put final.zip and units.txt here. diff --git a/runtime/device/android/wenet/app/src/main/cpp/wenet.cc b/runtime/device/android/wenet/app/src/main/cpp/wenet.cc index 1ea14958f..7c8e92a37 100644 --- a/runtime/device/android/wenet/app/src/main/cpp/wenet.cc +++ b/runtime/device/android/wenet/app/src/main/cpp/wenet.cc @@ -38,7 +38,7 @@ void init(JNIEnv* env, jobject, jstring jModelDir) { const char* pModelDir = env->GetStringUTFChars(jModelDir, nullptr); std::string modelPath = std::string(pModelDir) + "/final.zip"; - std::string dictPath = std::string(pModelDir) + "/words.txt"; + std::string dictPath = std::string(pModelDir) + "/units.txt"; auto model = std::make_shared(); model->Read(modelPath); LOG(INFO) << "model path: " << modelPath; diff --git a/runtime/device/android/wenet/app/src/main/java/com/mobvoi/wenet/MainActivity.java b/runtime/device/android/wenet/app/src/main/java/com/mobvoi/wenet/MainActivity.java index b7e574591..715170326 100644 --- a/runtime/device/android/wenet/app/src/main/java/com/mobvoi/wenet/MainActivity.java +++ b/runtime/device/android/wenet/app/src/main/java/com/mobvoi/wenet/MainActivity.java @@ -33,7 +33,7 @@ public class MainActivity extends AppCompatActivity { private static final int SAMPLE_RATE = 16000; // The sampling rate private static final int MAX_QUEUE_SIZE = 2500; // 100 seconds audio, 1 / 0.04 * 100 private static final List resource = Arrays.asList( - "final.zip", "words.txt", "ctc.ort", "decoder.ort", "encoder.ort" + "final.zip", "units.txt", "ctc.ort", "decoder.ort", "encoder.ort" ); private boolean startRecord = false;