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

pass model path

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. common.py +0 -1
app.py CHANGED
@@ -52,7 +52,8 @@ def get_client(user_id):
52
  FHEClient: The client API.
53
  """
54
  return FHEClient(
55
- key_dir=KEYS_PATH / f"seizure_detection_{user_id}",
 
56
  )
57
 
58
  def get_client_file_path(name, user_id):
 
52
  FHEClient: The client API.
53
  """
54
  return FHEClient(
55
+ model_path="ThomasCdnns/EEG-Seizure-Detection/resolve/main/seizure_detection_model-4.pth",
56
+ key_dir=KEYS_PATH / f"seizure_detection_{user_id}"
57
  )
58
 
59
  def get_client_file_path(name, user_id):
common.py CHANGED
@@ -6,7 +6,6 @@ from pathlib import Path
6
  REPO_DIR = Path(__file__).parent
7
 
8
  # This repository's main necessary folders
9
- FILTERS_PATH = REPO_DIR / "filters"
10
  KEYS_PATH = REPO_DIR / ".fhe_keys"
11
  CLIENT_TMP_PATH = REPO_DIR / "client_tmp"
12
  SERVER_TMP_PATH = REPO_DIR / "server_tmp"
 
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"