Ad download progress bar...

utils.py

@@ -24,6 +24,13 @@ class CustomDataSet(torch.utils.data.Dataset):
         image = PilImage.open(img_loc).convert("RGB")
         tensor_image = self.transform(image)
         return tensor_image
+        
+ 
+ class DownloadProgressBar(tqdm):
+    def update_to(self, b=1, bsize=1, tsize=None):
+        if tsize is not None:
+            self.total = tsize
+        self.update(b * bsize - self.n)
 
 
 def text_encoder(text, model, tokenizer):