diff --git a/.drone.yml b/.drone.yml
index 43e328a..dfb85fb 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -22,4 +22,4 @@ steps:
       - mkdir -p data
       - $mount_command
       - ls data/
-      - CI=False python facenet_transfer_learning.py
\ No newline at end of file
+      - CI=True python facenet_transfer_learning.py
\ No newline at end of file
diff --git a/facenet_transfer_learning.py b/facenet_transfer_learning.py
index 34cfd29..5edc5b6 100644
--- a/facenet_transfer_learning.py
+++ b/facenet_transfer_learning.py
@@ -151,7 +151,7 @@ for epoch in range(epochs):
     model_gpu.train()
     # makedirs(join(save_dir, 'pallets', str(epoch)), exist_ok=True)
 
-    for count, (images, labels) in enumerate(tqdm(dataloader['train'])):
+    for count, (images, labels) in enumerate(tqdm(dataloader['train'], disable=None)):
         if count == 1:
             image_pallets = plot_dataset(dataloader=(images, labels), col_len=6,
                                          label_text=image_folder['train'].classes)