Thomas Chardonnens commited on
Commit
fbd9a75
1 Parent(s): 6725add

fix model path with zips

Browse files
Files changed (2) hide show
  1. client_server_interface.py +2 -1
  2. 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 = 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"