Update index.html

index.html

@@ -1161,8 +1161,8 @@
             const generationEnd = performance.now();
             const totalTime = generationEnd - generationStart;
             const inputTokenCount = inputs.input_ids.dims[1];
-            // Handle tensor/array output properly
-            const totalTokenCount = Array.isArray(output) ? output.length : (output.size || output.length || 0);
+            // Handle tensor/array output properly - output is [batch_size, sequence_length]
+            const totalTokenCount = output.dims ? output.dims[1] : (Array.isArray(output[0]) ? output[0].length : output[0].size);
             const outputTokenCount = Math.max(0, totalTokenCount - inputTokenCount);
             const tps = outputTokenCount / (totalTime / 1000);