Update flash/core/data/utils.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Lightning-Universe · Mar 10, 2021 · a07f319 · a07f319
1 parent 5003f7d
commit a07f319
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/core/data/utils.py b/flash/core/data/utils.py
@@ -37,7 +37,7 @@ def download_file(url: str, path: str, verbose: bool = False) -> None:
 
     if not os.path.exists(local_filename):
         r = requests.get(url, stream=True)
-        file_size = int(r.headers['Content-Length']) if 'Content-Length' in r.headers else 0
+        file_size = int(r.headers.get('Content-Length', 0))
         chunk = 1
         chunk_size = 1024
         num_bars = int(file_size / chunk_size)