diff --git a/bert_gen.py b/bert_gen.py
index 193e768..4481471 100644
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -1,6 +1,6 @@
 import torch
 from torch.utils.data import DataLoader
-
+from multiprocessing import Pool
 import commons
 import utils
 from data_utils import TextAudioSpeakerLoader, TextAudioSpeakerCollate