fhe model client

common.py

@@ -5,6 +5,9 @@ from pathlib import Path
 # This repository's directory
 REPO_DIR = Path(__file__).parent
 
+CURRENT_DIR = Path(__file__).parent
+DEPLOYMENT_DIR = CURRENT_DIR / "deployment_files"
+
 # This repository's main necessary folders
 SEIZURE_DETECTION_MODEL_PATH = REPO_DIR / "seizure_detection/deployment"
 KEYS_PATH = REPO_DIR / ".fhe_keys"

server.py

@@ -5,11 +5,15 @@ from typing import List
 from fastapi import FastAPI, File, Form, UploadFile
 from fastapi.responses import JSONResponse, Response
 
-from common import SERVER_TMP_PATH
-from client_server_interface import FHEServer
+from concrete.ml.deployment import FHEModelServer
+
+from common import (
+    DEPLOYMENT_DIR,
+    SERVER_TMP_PATH
+)
 
 # Load the server object for seizure detection
-FHE_SERVER = FHEServer(model_path="ThomasCdnns/EEG-Seizure-Detection/resolve/main/seizure_detection_model-4.pth")
+FHE_SERVER = FHEModelServer(DEPLOYMENT_DIR)
 
 def get_server_file_path(name, user_id):
     """Get the correct temporary file path for the server.