update audit timestamps

app.py

@@ -10,7 +10,6 @@ from resources import Lead_Labels, entity_labels, set_start, audit_elapsedtime
 
 def main ():
     
-    
     rec = init_model_trans()
     ner = init_model_ner() #async
     
@@ -23,9 +22,6 @@ def main ():
     print("Render UI")
     wav_audio_data = st_audiorec()
 
-    if wav_audio_data is not None:
-        print("Loading data...")
-    
     if wav_audio_data is not None and rec is not None:
         print("Loading data...")
         st.audio(wav_audio_data, format='audio/wav')
@@ -36,7 +32,6 @@ def main ():
 
 def init_model_trans ():
     print("Initiating transcription model...")
-    func_name = "init_model_trans"
     start = set_start()
 
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -64,28 +59,31 @@ def init_model_trans ():
         device=device,
     )
     print(f'Init model successful: {model}' )
-    audit_elapsedtime(function="init_model_trans", start=start)
+    audit_elapsedtime(function="Initiating transcription model", start=start)
     return pipe
 
 def init_model_ner():
     print("Initiating NER model...")
     start = set_start()
     model = GLiNER.from_pretrained("urchade/gliner_multi")
-    audit_elapsedtime(function="init_model_ner", start=start)
+    audit_elapsedtime(function="Initiating NER model", start=start)
     return model
 
 def transcribe (audio_sample: bytes, pipe) -> str:
-
+    start = set_start()
     # dataset = load_dataset("distil-whisper/librispeech_long", "clean", split="validation")
     # sample = dataset[0]["audio"]
     result = pipe(audio_sample)
+    audit_elapsedtime(function="Transcription", start=start)
     print(result)
     
     st.write('trancription: ', result["text"])
     return result["text"]
 
 def get_entity_labels(model: GLiNER, text: str, labels: list): #-> Lead_labels:
+    start = set_start()
     entities = model.predict_entities(text, labels)
+    audit_elapsedtime(function="Retreiving entity labels from text", start=start)
 
     for entity in entities:
         print(entity["text"], "=>", entity["label"])