WenqingZhang commited on
Commit
45e9881
·
verified ·
1 Parent(s): 39cb02e

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +2 -2
server.py CHANGED
@@ -10,7 +10,7 @@ current_dir = Path(__file__).parent
10
 
11
  # Load the model
12
  fhe_model = FHEModelServer("deployment/financial_rating")
13
- fhe_legal_model = FHEModelServer("deployment/legal_rating")
14
  class PredictRequest(BaseModel):
15
  evaluation_key: str
16
  encrypted_encoding: str
@@ -25,7 +25,7 @@ def root():
25
 
26
  @app.post("/predict_sentiment")
27
  def predict_sentiment(query: PredictRequest):
28
- fhe_model = FHEModelServer("deployment/financial_rating")
29
 
30
  encrypted_encoding = base64.b64decode(query.encrypted_encoding)
31
  evaluation_key = base64.b64decode(query.evaluation_key)
 
10
 
11
  # Load the model
12
  fhe_model = FHEModelServer("deployment/financial_rating")
13
+ #fhe_legal_model = FHEModelServer("deployment/legal_rating")
14
  class PredictRequest(BaseModel):
15
  evaluation_key: str
16
  encrypted_encoding: str
 
25
 
26
  @app.post("/predict_sentiment")
27
  def predict_sentiment(query: PredictRequest):
28
+
29
 
30
  encrypted_encoding = base64.b64decode(query.encrypted_encoding)
31
  evaluation_key = base64.b64decode(query.evaluation_key)