Thomas Chardonnens
commited on
Commit
•
fbd9a75
1
Parent(s):
6725add
fix model path with zips
Browse files- client_server_interface.py +2 -1
- common.py +1 -0
client_server_interface.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
"Client-server interface custom implementation for seizure detection models."
|
2 |
|
|
|
3 |
from concrete import fhe
|
4 |
|
5 |
from seizure_detection import SeizureDetector
|
@@ -85,7 +86,7 @@ class FHEClient:
|
|
85 |
model_path (Path): The path to the directory where the circuit is saved.
|
86 |
key_dir (Path): The path to the directory where the keys are stored. Default to None.
|
87 |
"""
|
88 |
-
self.model_path =
|
89 |
self.key_dir = key_dir
|
90 |
|
91 |
# If model_path does not exist raise
|
|
|
1 |
"Client-server interface custom implementation for seizure detection models."
|
2 |
|
3 |
+
from common import SEIZURE_DETECTION_MODEL_PATH
|
4 |
from concrete import fhe
|
5 |
|
6 |
from seizure_detection import SeizureDetector
|
|
|
86 |
model_path (Path): The path to the directory where the circuit is saved.
|
87 |
key_dir (Path): The path to the directory where the keys are stored. Default to None.
|
88 |
"""
|
89 |
+
self.model_path = SEIZURE_DETECTION_MODEL_PATH
|
90 |
self.key_dir = key_dir
|
91 |
|
92 |
# If model_path does not exist raise
|
common.py
CHANGED
@@ -6,6 +6,7 @@ from pathlib import Path
|
|
6 |
REPO_DIR = Path(__file__).parent
|
7 |
|
8 |
# This repository's main necessary folders
|
|
|
9 |
KEYS_PATH = REPO_DIR / ".fhe_keys"
|
10 |
CLIENT_TMP_PATH = REPO_DIR / "client_tmp"
|
11 |
SERVER_TMP_PATH = REPO_DIR / "server_tmp"
|
|
|
6 |
REPO_DIR = Path(__file__).parent
|
7 |
|
8 |
# This repository's main necessary folders
|
9 |
+
SEIZURE_DETECTION_MODEL_PATH = REPO_DIR / "seizure_detection/deployment"
|
10 |
KEYS_PATH = REPO_DIR / ".fhe_keys"
|
11 |
CLIENT_TMP_PATH = REPO_DIR / "client_tmp"
|
12 |
SERVER_TMP_PATH = REPO_DIR / "server_tmp"
|