From 5daf272d855e6ac069a0b4500d0adc322bbac3e3 Mon Sep 17 00:00:00 2001
From: wangxiao1021 <zko1021@foxmail.com>
Date: Mon, 10 Feb 2020 21:50:01 +0800
Subject: [PATCH] fix download

---
 paddlepalm/_downloader.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddlepalm/_downloader.py b/paddlepalm/_downloader.py
index c0dc339..dfa3a69 100644
--- a/paddlepalm/_downloader.py
+++ b/paddlepalm/_downloader.py
@@ -15,6 +15,7 @@
 
 from __future__ import print_function
 import os
+import requests
 import tarfile
 import shutil
 try:
@@ -67,8 +68,7 @@ def _download(item, scope, path, silent=False, convert=False):
     
     # copy to local
     def _chunk_read(response, url, chunk_size = 16 * 1024, report_hook = None):
-        total_size = response.info().getheader('Content-Length').strip()
-        total_size = int(total_size)
+        total_size = int(requests.head(url).headers['Content-Length'])
         bytes_so_far = 0
         with open("%s" % filename, "wb") as f:
             while 1:
-- 
GitLab