diff --git a/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java b/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java index f2529eb6b86..997d9f87f92 100644 --- a/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java +++ b/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java @@ -119,6 +119,10 @@ private String memoryInfo() { private void modelDialog() { String[] pteFiles = listLocalFile("/data/local/tmp/llama/", ".pte"); String[] binFiles = listLocalFile("/data/local/tmp/llama/", ".bin"); + String[] modelFiles = listLocalFile("/data/local/tmp/llama/", ".model"); + String[] tokenizerFiles = new String[binFiles.length + modelFiles.length]; + System.arraycopy(binFiles, 0, tokenizerFiles, 0, binFiles.length); + System.arraycopy(modelFiles, 0, tokenizerFiles, binFiles.length, modelFiles.length); AlertDialog.Builder modelPathBuilder = new AlertDialog.Builder(this); modelPathBuilder.setTitle("Select model path"); AlertDialog.Builder tokenizerPathBuilder = new AlertDialog.Builder(this); @@ -134,10 +138,10 @@ private void modelDialog() { }); tokenizerPathBuilder.setSingleChoiceItems( - binFiles, + tokenizerFiles, -1, (dialog, item) -> { - mTokenizerFilePath = binFiles[item]; + mTokenizerFilePath = tokenizerFiles[item]; Runnable runnable = new Runnable() { @Override