Spaces:
Running
Running
File size: 833 Bytes
96134ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import os
import tqdm
import requests
def HF_download_file(url, output_path=None):
url = url.replace("/blob/", "/resolve/").strip()
if output_path is None:
output_path = os.path.basename(url)
elif os.path.isdir(output_path):
output_path = os.path.join(output_path, os.path.basename(url))
response = requests.get(url, stream=True)
if response.status_code == 200:
progress_bar = tqdm.tqdm(total=int(response.headers.get("content-length", 0)), desc=os.path.basename(url), unit="byte")
with open(output_path, "wb") as f:
for chunk in response.iter_content(chunk_size=1024*1024):
progress_bar.update(len(chunk))
f.write(chunk)
progress_bar.close()
return output_path
else:
raise ValueError(response.status_code)
|