diff --git a/src/infer.py b/src/infer.py
index 1726fd8..91678ef 100644
--- a/src/infer.py
+++ b/src/infer.py
@@ -5,7 +5,6 @@
 from torch.nn.utils.rnn import pad_sequence
 from torchdiffeq import odeint
 from safetensors.torch import load_file
-import IPython.display as ipd
 
 # Import F5-TTS modules
 from f5_tts.model import CFM, UNetT, DiT