Show time for inference

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 import allin1
+import time
 
 from pathlib import Path
 
@@ -44,6 +45,9 @@ CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
 
 
 def analyze(path):
+  #Measure time for inference
+  start = time.time()
+    
   path = Path(path)
   result = allin1.analyze(
     path,
@@ -65,7 +69,11 @@ def analyze(path):
   )
   sonif_path = Path(f'./sonif/{path.stem}.sonif{path.suffix}').resolve().as_posix()
 
-  return result.bpm, fig, sonif_path
+  #Measure time for inference
+  end = time.time()
+  elapsed_time = end-start
+
+  return result.bpm, fig, sonif_path, elapsed_time
 
 
 with gr.Blocks() as demo:
@@ -89,6 +97,7 @@ with gr.Blocks() as demo:
       show_download_button=False,
       scale=9,
     )
+    elapsed_time = gr.Textbox(label='Overall inference time', scale=1)
   
   #gr.Examples(
   #  examples=[
@@ -104,7 +113,7 @@ with gr.Blocks() as demo:
   button.click(
     fn=analyze,
     inputs=input_audio_path,
-    outputs=[output_bpm, output_viz, output_sonif],
+    outputs=[output_bpm, output_viz, output_sonif, elapsed_time],
     api_name='analyze',
   )