romanbredehoft-zama's picture
Compile deployment files on python 3.10
fe8d446
import pandas as pd
from concrete.ml.deployment import FHEModelDev
from concrete.ml.common.serialization.loaders import load
import shutil
from pathlib import Path
script_dir = Path(__file__).parent
DEPLOYMENT_DIR = script_dir / "deployment"
print("Compiling the model...")
with (DEPLOYMENT_DIR / "serialized_model").open("r") as file:
model = load(file)
# Load the data from the csv file to be used for compilation
data = pd.read_csv(DEPLOYMENT_DIR / "samples_for_compilation.csv", index_col=0).values
# Compile the model
model.compile(data)
dev_model_path = DEPLOYMENT_DIR / "sentiment_fhe_model"
# Delete the deployment folder if it exist
if dev_model_path.is_dir():
shutil.rmtree(dev_model_path)
fhe_api = FHEModelDev(
model=model, path_dir=dev_model_path
)
fhe_api.save(via_mlir=True)
print("Done!")