diff --git a/train_.py b/train_.py
index 3f56a62..46c369a 100644
--- a/train_.py
+++ b/train_.py
@@ -9,7 +9,7 @@
 import matplotlib.pyplot as plt
 import os,time
 from PIL import Image
-# from tqdm import tqdm
+from tqdm import tqdm
 import numpy as np
 
 class ae_gan():