Update app.py

app.py

@@ -24,9 +24,11 @@ def classify_audio(filepath):
      to
      {"country":  0.8339303731918335, "rock":0.11914275586605072}
     """
+    start_time = timer()
     preds = pipe(filepath)
     # preds = predict_trunc(filepath)
     outputs = {}
+    pred_time = round(timer() - start_time, 5)
     for p in preds:
         outputs[p["label"]] = p["score"]
     return outputs
@@ -43,7 +45,7 @@ filenames = [[f"./{f}"] for f in filenames]
 demo = gr.Interface(
     fn=classify_audio,
     inputs=gr.Audio(type="filepath"),
-    outputs=gr.outputs.Label(),
+    outputs=[gr.outputs.Label(),gr.Number(label="Prediction time (s)")]
     title=title,
     description=description,
     examples=filenames,