import base64

with open("models/best_residue_model.pt", "rb") as f:
    data = base64.b64encode(f.read()).decode()

with open("models/model_b64.py", "w") as f:
    f.write('MODEL_DATA = "' + data + '"')

print("Done!")