diff --git a/predict.py b/predict.py
index 99890fd..949f189 100644
--- a/predict.py
+++ b/predict.py
@@ -22,7 +22,7 @@
     plot_das,
     plot_phasenet,
 )
-from tqdm.auto import tqdm
+import tqdm
 
 # mp.set_start_method("spawn", force=True)
 matplotlib.use("agg")