| import onnx | |
| import pandas as pd | |
| from concrete.ml.deployment import FHEModelDev, FHEModelClient | |
| from concrete.ml.common.serialization.loaders import load | |
| from concrete.ml.onnx.convert import get_equivalent_numpy_forward_from_onnx_tree | |
| import json | |
| import os | |
| 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!") | |