fix

Author: kirklandsign

examples/demo-apps/android/LlamaDemo/app/src/androidTest/java/com/example/executorchllamademo/PerfTest.java

@@ -66,13 +66,16 @@ public void onResult(String result) {
   }
 
   @Override
-  public void onStats(String result) throws JSONException {
-    JSONObject jsonObject = new JSONObject(result);
-    int numGeneratedTokens = jsonObject.getInt("num_generated_tokens");
-    int inferenceEndMs = jsonObject.getInt("inference_end_ms");
-    int promptEvalEndMs = jsonObject.getInt("prompt_eval_end_ms");
-    float tps = (float) numGeneratedTokens / (inferenceEndMs - promptEvalEndMs) * 1000;
-    tokensPerSecond.add(tps);
+  public void onStats(String result) {
+    try {
+      JSONObject jsonObject = new JSONObject(result);
+      int numGeneratedTokens = jsonObject.getInt("num_generated_tokens");
+      int inferenceEndMs = jsonObject.getInt("inference_end_ms");
+      int promptEvalEndMs = jsonObject.getInt("prompt_eval_end_ms");
+      float tps = (float) numGeneratedTokens / (inferenceEndMs - promptEvalEndMs) * 1000;
+      tokensPerSecond.add(tps);
+    } catch (JSONException e) {
+    }
   }
 
   private void report(final String metric, final Float value) {

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/LlmBenchmarkActivity.java

@@ -81,7 +81,7 @@ public void onModelLoaded(int status) {
   public void onTokenGenerated(String token) {}
 
   @Override
-  public void onStats(String result) {
+  public void onStats(String stats) {
     float tps = 0;
     try {
       JSONObject jsonObject = new JSONObject(stats);