in scripts/generate-registry.py [0:0]
def get_meta(model_path, model_type):
meta = {}
for file_path in glob.glob(f"{model_path}/*.gz"):
name = os.path.basename(file_path)[:-3]
size = os.path.getsize(file_path)
with gzip.open(file_path, "rb") as f:
bytes = f.read()
hash = hashlib.sha256(bytes).hexdigest()
key = [key for key in KEYS if name.startswith(key)][0]
meta[key] = {
"name": name,
"size": len(bytes),
"estimatedCompressedSize": size,
"expectedSha256Hash": hash,
"modelType": model_type,
}
return meta