Update index.html

index.html

@@ -30,6 +30,8 @@
         let speaking = false
         let total_infer_time=0
         let count_infer=0
+        let loaded_model_name
+        let load_time
         async function main(model_name) {
           if (typeof model_name !== 'string') {//via button click
                 model_name ="en001_ep6399_univ_simplify"
@@ -44,6 +46,7 @@
             speaking = true
             console.log("main called")
             if(!matcha_tts_raw){
+                const load_startTime = performance.now();
                 matcha_tts_raw = new MatchaTTSRaw()
                 console.time("load model");
                 const model_path = `./models/matcha-tts/${model_name}.onnx`
@@ -51,10 +54,15 @@
                 await matcha_tts_raw.load_model(model_path,{ executionProviders: ['webgpu','wasm'] });
                 
                 console.timeEnd("load model");
-                
+
+                load_time = load_startTime - performance.now()
+                loaded_model_name = model_name
+              
                 let cmudictReady = loadCmudict(cmudict,'./dictionaries/cmudict-0.7b')
                 await cmudictReady
 
+                
+                update_infer_bench2()
             }else{
                 console.log("session exist skip load model")
             }
@@ -79,7 +87,7 @@
               const infer_time = endTime-startTime
               total_infer_time+=infer_time
               count_infer += 1
-              update_infer_bench1()
+              update_infer_bench2()
                 webWavPlay(result)
               
                
@@ -88,9 +96,15 @@
             speaking = false
         }
       function update_infer_bench1(){
+      
+      const text = `${loaded_model_name} load time ${load_time.toFixed(2)} ms`;
+      document.getElementById('result2').innerText=text
+      }
+      
+      function update_infer_bench2(){
       const avg = total_infer_time/count_infer
       const text = `Infer Count ${count_infer} avg infer-time ${avg.toFixed(2)} ms`;
-      document.getElementById('result').innerText=text
+      document.getElementById('result2').innerText=text
       }
         function update_range(){
             const value = document.getElementById('spks').value
@@ -143,9 +157,11 @@
       document.getElementById('buttons').appendChild(create_button("en001","en001_ep6399_univ_simplify"))
       document.getElementById('buttons').appendChild(create_button("en001-quantized","en001_ep6399_univ_simplify_q8"))
     </script>
+<div id="result1"></div>
+  <br>
 　<div id="buttons"></div>
-  
-    <div id="result"></div>
+<br>
+    <div id="result2"></div>
   <br><br>
     <input type="text" id="textInput"  value ="Hello Huggingface." placeholder="Enter some text here...">