Update app.py

app.py

@@ -228,7 +228,7 @@ def tse(test_wav, enroll_wav):
         sample = sample.squeeze()
         x_hat = geco_model.to_audio(sample.squeeze(), min_leng)
         x_hat = x_hat * norm_factor / x_hat.abs().max()
-        x_hat = x_hat.detach().cpu()
+        x_hat = x_hat.detach().cpu().squeeze().numpy()
 
         end_time = time.time()
         audio_len = x_hat.shape[-1] / 16000