Update app.py

app.py

@@ -3,6 +3,7 @@ import torchaudio
 from sgmse.model import ScoreModel
 import gradio as gr
 from sgmse.util.other import pad_spec
+import time  # Import the time module
 
 # Define parameters based on the argparse configuration in enhancement.py
 args = {
@@ -20,7 +21,7 @@ args = {
 model = ScoreModel.load_from_checkpoint(args["ckpt"])
 
 def enhance_speech(audio_file):
-    start_time = time.time()
+    start_time = time.time()  # Start the timer
 
     # Load and process the audio file
     y, sr = torchaudio.load(audio_file)
@@ -57,7 +58,6 @@ def enhance_speech(audio_file):
     print(f"Processed audio in {time.time() - start_time:.2f}s")
     return output_file
 
-
 # Gradio interface setup
 inputs = gr.Audio(label="Input Audio", type="filepath")
 outputs = gr.Audio(label="Output Audio", type="filepath")