Update app.py

app.py

@@ -5,7 +5,7 @@ import tensorflow as tf
 import sys
 import numpy as np
 import csv
-import datetime
+import datetime, time
 import joblib
 from huggingface_hub import hf_hub_download
 
@@ -39,11 +39,14 @@ def process(prompt, lang):
     print("Text input : ", prompt)
     print('*************')
     print()
-    
+    a=time.time()
     #try :
     embed_query  = get_predict(encoder_text, prompt, preprocess_model, model)
+    print("Embed time : ", time.time() - a)
     do_normalize(embed_query)
     D, I = get_distance(index, embed_query, TOP)
+    print("Search + Embed time : ", time.time() - a)
+    
     #print(I)
     #print(D)
     #print("----")
@@ -70,6 +73,7 @@ def process(prompt, lang):
         out.append(audio_names[I[0][i]].split('.')[0])
         out.append(get_url_myma(I[0][i], audio_names, url_dict))
 
+    print("Total time : ", time.time() - a)
     return out
     '''return [output_csv,
             audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict),