print elapsedtime

app.py

@@ -25,19 +25,24 @@ def main ():
     audit_elapsedtime(function="Rendering UI", start=start_render)
 
     if wav_audio_data is not None:
+        start_loading = set_start()
+
         s2t = init_model_trans()
-        # ner = init_model_ner()
+        ner = init_model_ner()
+
         print("Loading data...")
-        start_loading = set_start()
         #st.audio(wav_audio_data, format='audio/wav')
         original = transcribe(wav_audio_data, s2t)
         st.write(f"Transcription: {original}")
         translation = translate(original)
         st.write(f"Translation: {translation}")
 
-    #     if text is not None and ner is not None:    
-    #         st.write('Entities: ', get_entity_labels(model=ner, text=text))
-    #     audit_elapsedtime(function="Loading data", start=start_loading)
+        if translation is not None and ner is not None:    
+            st.write('Entities: ', get_entity_labels(model=ner, text=translation))
+        loading_elapsedtime = audit_elapsedtime(function="Loading data", start=start_loading)
+
+        st.write(f"Total elapsed time: {loading_elapsedtime} secs")
+
 
 if __name__ == "__main__":
     print("IN __name__")

nameder.py

@@ -20,4 +20,5 @@ def get_entity_labels(model: GLiNER, text: str): #-> Lead_labels:
 
     for entity in entities:
         print(entity["text"], "=>", entity["label"])
+
     return entities

resources.py

@@ -23,8 +23,10 @@ entity_labels = [
 def set_start () -> time:
     return time.time()
 
-def audit_elapsedtime(function: str, start: time) -> None:
+def audit_elapsedtime(function: str, start: time):
     end = time.time()
+    elapsedtime = end-start
     print("------------------")
-    print(f"[{function}] Elapsed time: {end-start}")
-    print("------------------")
+    print(f"[{function}] Elapsed time: {elapsedtime}")
+    print("------------------")
+    return elapsedtime

speech2text.py

@@ -33,7 +33,7 @@ def init_model_trans ():
         device=device,
     )
     print(f'Init model successful')
-
+    audit_elapsedtime(function="Init transc model", start=start)
     return pipe
 
 def transcribe (audio_sample: bytes, pipe) -> str: