def get_file_path(self):
"""
Return local file path of downloaded model files
"""
for filename in self.files:
cached_file_path, need_downloading = cached_path(
self.base_path + filename, self.download_path)
if filename.endswith("zip"):
if need_downloading:
print("Unzipping...")
uncompress_file(cached_file_path,
outpath=self.download_path)
print("Done.")
return self.download_path
def get_model_files(self):
"""
Return individual file names of downloaded models
"""
for fileName in self.files:
cached_file_path, need_downloading = cached_path(
self.base_path + fileName, self.download_path)
if fileName.endswith("zip"):
if need_downloading:
print("Unzipping...")
uncompress_file(cached_file_path,
outpath=self.download_path)
print("Done.")
self.model_files.extend(zipfile_list(cached_file_path))
else:
self.model_files.extend([fileName])
return self.model_files
