Update server.py

server.py

@@ -6,11 +6,14 @@ from pydantic import BaseModel
 import base64
 from pathlib import Path
 
+current_dir = Path(__file__).parent
+
+
 current_dir = Path(__file__).parent
 
 # Load the model
 fhe_model = FHEModelServer("deployment/financial_rating")
-fhe_legal_model = FHEModelServer("deployment/legal_rating")
+
 class PredictRequest(BaseModel):
     evaluation_key: str
     encrypted_encoding: str
@@ -25,8 +28,6 @@ def root():
 
 @app.post("/predict_sentiment")
 def predict_sentiment(query: PredictRequest):
-    fhe_model = FHEModelServer("deployment/financial_rating")
-
     encrypted_encoding = base64.b64decode(query.encrypted_encoding)
     evaluation_key = base64.b64decode(query.evaluation_key)
     prediction = fhe_model.run(encrypted_encoding, evaluation_key)