diff --git a/train_class.py b/train_class.py
index 0c4cfd6..5afd6b7 100644
--- a/train_class.py
+++ b/train_class.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():