download: fix for text files without content-length

nim65s · Oct 30, 2023 · 0c2846d · 0c2846d
1 parent 2d299da
commit 0c2846d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/happypose/toolbox/utils/download.py b/happypose/toolbox/utils/download.py
@@ -391,7 +391,9 @@ async def download_file(self, download_path, local_path):
         local_path.parent.mkdir(parents=True, exist_ok=True)
         with local_path.open("wb") as f:
             async with self.client.stream("GET", download_path) as r:
-                total = int(r.headers["Content-Length"])
+                total = None
+                if "Content-Length" in r.headers:
+                    total = int(r.headers["Content-Length"])
                 with tqdm(
                     desc=local_path.name,
                     total=total,