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