Update server.py

server.py

@@ -34,7 +34,8 @@ def predict_sentiment(query: PredictRequest):
     # Encode base64 the prediction
     encoded_prediction = base64.b64encode(prediction).decode()
     return {"encrypted_prediction": encoded_prediction}
-    
+'''
+  
 @app.post("/predict_legal")
 def predict_legal(query: PredictRequest):
     fhe_legal_model = FHEModelServer("deployment/legal_rating")
@@ -46,3 +47,4 @@ def predict_legal(query: PredictRequest):
     # Encode base64 the prediction
     encoded_prediction = base64.b64encode(prediction).decode()
     return {"encrypted_prediction": encoded_prediction}
+'''