Thomas Chardonnens commited on
Commit
d313727
1 Parent(s): b831724

update repo server model

Browse files
Files changed (1) hide show
  1. server.py +2 -2
server.py CHANGED
@@ -5,11 +5,11 @@ from typing import List
5
  from fastapi import FastAPI, File, Form, UploadFile
6
  from fastapi.responses import JSONResponse, Response
7
 
8
- from common import SERVER_TMP_PATH
9
  from client_server_interface import FHEServer
10
 
11
  # Load the server object for seizure detection
12
- FHE_SERVER = FHEServer(model_path="ThomasCdnns/EEG-Seizure-Detection/resolve/main/seizure_detection_model-4.pth")
13
 
14
  def get_server_file_path(name, user_id):
15
  """Get the correct temporary file path for the server.
 
5
  from fastapi import FastAPI, File, Form, UploadFile
6
  from fastapi.responses import JSONResponse, Response
7
 
8
+ from common import SERVER_TMP_PATH, SEIZURE_DETECTION_MODEL_PATH
9
  from client_server_interface import FHEServer
10
 
11
  # Load the server object for seizure detection
12
+ FHE_SERVER = FHEServer(model_path=SEIZURE_DETECTION_MODEL_PATH)
13
 
14
  def get_server_file_path(name, user_id):
15
  """Get the correct temporary file path for the server.