Thomas Chardonnens
commited on
Commit
•
6725add
1
Parent(s):
4d389e0
pass model path
Browse files
app.py
CHANGED
@@ -52,7 +52,8 @@ def get_client(user_id):
|
|
52 |
FHEClient: The client API.
|
53 |
"""
|
54 |
return FHEClient(
|
55 |
-
|
|
|
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"
|