chore: see the size of the key

creating_models.py

@@ -1,7 +1,9 @@
 import shutil
+import sys
 from pathlib import Path
 
 from concrete.ml.deployment import FHEModelDev
+from concrete.ml.deployment import FHEModelClient
 
 
 def compile_and_make_it_deployable(model_dev, X_train):
@@ -16,6 +18,15 @@ def compile_and_make_it_deployable(model_dev, X_train):
     fhemodel_dev = FHEModelDev(path_to_model, model_dev)
     fhemodel_dev.save(via_mlir=True)
 
+    # To see the size of the key
+    fhemodel_client = FHEModelClient(path_to_model)
+
+    # Generate the keys
+    fhemodel_client.generate_private_and_evaluation_keys()
+    evaluation_keys = fhemodel_client.get_serialized_evaluation_keys()
+
+    print(f"Your keys will be {sys.getsizeof(evaluation_keys) / 1024 / 1024}-megabytes long")
+
 
 # This is the spam classifier. Taken from https://github.com/zama-ai/concrete-ml/blob/main/docs/advanced_examples/DecisionTreeClassifier.ipynb
 import numpy