downloading models from github releases

pull/2/head
Eren Gölge 2021-03-10 11:09:01 +01:00
parent fc19411ac6
commit 599149a7e5
1 changed files with 6 additions and 2 deletions

View File

@ -3,6 +3,7 @@ import json
import os
import zipfile
from pathlib import Path
from shutil import copyfile
import gdown
import requests
@ -130,11 +131,14 @@ class ModelManager(object):
@staticmethod
def _download_zip_file(file_url, output):
"""Download the target zip file and extract the files
to a folder with the same name as the zip file."""
"""Download the github releases"""
r = requests.get(file_url)
z = zipfile.ZipFile(io.BytesIO(r.content))
z.extractall(output)
for file_path in z.namelist()[1:]:
src_path = os.path.join(output, file_path)
dst_path = os.path.join(output, os.path.basename(file_path))
copyfile(src_path, dst_path)
@staticmethod
def _check_dict_key(my_dict, key):