Update app.py

app.py

@@ -44,14 +44,17 @@ def transcribe(mic_input, upl_input, model_type):
     else:
         audio = upl_input
     time.sleep(3)
+    st_time = time.time()
     if model_type == 'NoisyFinetuned':
         text = p2(audio)["text"]
     elif model_type == 'CleanFinetuned':
         text = p3(audio)["text"]
     else:
         text = p1(audio)["text"]
+    end_time = time.time()
     # state = text + " "
-    return text
+    time_taken = (end_time - st_time) / 60
+    return text, time_taken
 
 
 
@@ -79,7 +82,7 @@ def transcribe(mic_input, upl_input, model_type):
 # demo.launch()
 
 def clear_inputs_and_outputs():
-    return [None, None, "CleanFinetuned", None]
+    return [None, None, "CleanFinetuned", None, None]
 
 # Main function
 if __name__ == "__main__":
@@ -111,7 +114,10 @@ if __name__ == "__main__":
 
             # Outputs
             with gr.Column():
-                lbl_output = gr.Label(label="Top Predictions")
+                lbl_output = gr.Label(label="Transcription")
+
+            with gr.Column():
+                time_output = gr.Label(label="Time Taken (in sec)")
                 # with gr.Group():
                 #     gr.Markdown("<center>Prediction per time slot</center>")
                 #     plt_output = gr.Plot(
@@ -126,7 +132,7 @@ if __name__ == "__main__":
                 os.path.join(os.path.dirname(__file__), "audio/sample2.wav"),
             ],
             upl_input,
-            lbl_output,
+            [lbl_output, time_output],
             transcribe
         )
         # Credits
@@ -142,12 +148,12 @@ if __name__ == "__main__":
         clr_btn.click(
             fn=clear_inputs_and_outputs,
             inputs=[],
-            outputs=[mic_input, upl_input, model_type, lbl_output],
+            outputs=[mic_input, upl_input, model_type, lbl_output, time_output],
         )
         prd_btn.click(
             fn=transcribe,
             inputs=[mic_input, upl_input, model_type],
-            outputs=[lbl_output],
+            outputs=[lbl_output, time_output],
         )
 
     demo.launch(debug=True)