Upload server.py

server.py

@@ -9,7 +9,7 @@ from pathlib import Path
 current_dir = Path(__file__).parent
 
 # Load the model
-fhe_model_HLM = FHEModelServer(
+fhe_model = FHEModelServer(
     Path.joinpath(current_dir, "fhe_model")
 )
 class PredictRequest(BaseModel):
@@ -28,10 +28,10 @@ def root():
 
 
 @app.post("/predict")
-def predict_HLM(query: PredictRequest):
+def predict(query: PredictRequest):
     encrypted_encoding = base64.b64decode(query.encrypted_encoding)
     evaluation_key = base64.b64decode(query.evaluation_key)
-    prediction = fhe_model_HLM.run(encrypted_encoding, evaluation_key)
+    prediction = fhe_model.run(encrypted_encoding, evaluation_key)
 
     # Encode base64 the prediction
     encoded_prediction = base64.b64encode(prediction).decode()