Update app.py

app.py

@@ -5,7 +5,7 @@ import zipfile
 import natsort
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 from pathlib import Path
-from tqdm import tqdm
+import stqdm
 import streamlit as st
 from jax import numpy as jnp
 import transformers
@@ -41,10 +41,13 @@ def download_images():
             total_size_in_bytes= int(response.headers.get('content-length', 0))
             block_size = 1024 #1 Kb
             progress_bar = tqdm(total=total_size_in_bytes, unit='iB', unit_scale=True)
+            content = io.BytesIO()
             for data in response.iter_content(block_size):
                 progress_bar.update(len(data))
+                content.write(data)
             progress_bar.close()
-            z = zipfile.ZipFile(io.BytesIO(response.content))
+            content.seek()
+            z = zipfile.ZipFile(content)
             print("Extracting the dataset...")
             z.extractall(path=img_folder)
     print("Done.")