| 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) | |