fix performance log

app.py

@@ -42,9 +42,9 @@ def run_conversion(audio_in):
     source = torch.mean(source, dim=0).unsqueeze(0)
     source = source.unsqueeze(0)
 
-    time_start = time.perf_counter()
-    
     with torch.inference_mode():
+        time_start = time.perf_counter()
+        
         # Extract speech units
         units = hubert.units(source)
         # Generate target spectrogram
@@ -54,12 +54,13 @@ def run_conversion(audio_in):
         
         result = target.squeeze().cpu().multiply(32767).to(torch.int16).numpy()
 
-        return (16000, result)
+        time_end = time.perf_counter()
+        time_elapsed = time_end - time_start
         
-    time_end = time.perf_counter()
-    time_elapsed = time_end - time_start
-    
-    print(f"Conversion finished in {time_elapsed} Seconds")
+        print(f"Conversion finished in {time_elapsed} Seconds")
+        
+        return (16000, result)
+
 
 
 with gr.Blocks() as demo: