awss / models /reconstruct_models.py
F16Sam's picture
Upload 7 files
0bc891a verified
import os
import subprocess
MODEL_DIR = os.path.dirname(__file__)
def reassemble_chunks():
bio_chunks = sorted([f for f in os.listdir(MODEL_DIR) if f.startswith("layer2bio_")])
non_chunks = sorted([f for f in os.listdir(MODEL_DIR) if f.startswith("layer2non_")])
bio_target = os.path.join(MODEL_DIR, "layer2bio_cnn.keras")
non_target = os.path.join(MODEL_DIR, "layer2non_cnn.keras")
# Reassemble only if not already present
if not os.path.exists(bio_target):
with open(bio_target, 'wb') as wfd:
for chunk in bio_chunks:
with open(os.path.join(MODEL_DIR, chunk), 'rb') as fd:
wfd.write(fd.read())
print("✅ Reconstructed layer2bio_cnn.keras")
if not os.path.exists(non_target):
with open(non_target, 'wb') as wfd:
for chunk in non_chunks:
with open(os.path.join(MODEL_DIR, chunk), 'rb') as fd:
wfd.write(fd.read())
print("✅ Reconstructed layer2non_cnn.keras")
if __name__ == "__main__":
reassemble_chunks()