Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import zipfile
|
|
5 |
import natsort
|
6 |
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
7 |
from pathlib import Path
|
8 |
-
|
9 |
import streamlit as st
|
10 |
from jax import numpy as jnp
|
11 |
import transformers
|
@@ -41,10 +41,13 @@ def download_images():
|
|
41 |
total_size_in_bytes= int(response.headers.get('content-length', 0))
|
42 |
block_size = 1024 #1 Kb
|
43 |
progress_bar = tqdm(total=total_size_in_bytes, unit='iB', unit_scale=True)
|
|
|
44 |
for data in response.iter_content(block_size):
|
45 |
progress_bar.update(len(data))
|
|
|
46 |
progress_bar.close()
|
47 |
-
|
|
|
48 |
print("Extracting the dataset...")
|
49 |
z.extractall(path=img_folder)
|
50 |
print("Done.")
|
|
|
5 |
import natsort
|
6 |
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
7 |
from pathlib import Path
|
8 |
+
import stqdm
|
9 |
import streamlit as st
|
10 |
from jax import numpy as jnp
|
11 |
import transformers
|
|
|
41 |
total_size_in_bytes= int(response.headers.get('content-length', 0))
|
42 |
block_size = 1024 #1 Kb
|
43 |
progress_bar = tqdm(total=total_size_in_bytes, unit='iB', unit_scale=True)
|
44 |
+
content = io.BytesIO()
|
45 |
for data in response.iter_content(block_size):
|
46 |
progress_bar.update(len(data))
|
47 |
+
content.write(data)
|
48 |
progress_bar.close()
|
49 |
+
content.seek()
|
50 |
+
z = zipfile.ZipFile(content)
|
51 |
print("Extracting the dataset...")
|
52 |
z.extractall(path=img_folder)
|
53 |
print("Done.")
|