Upload handler.py

handler.py

@@ -3,6 +3,11 @@ import numpy as np
 from concrete.ml.deployment import FHEModelServer
 
 
+def from_json(python_object):
+    if "__class__" in python_object:
+        return bytes(python_object["__value__"])
+
+
 class EndpointHandler:
     def __init__(self, path="benoit.model"):
 
@@ -19,10 +24,10 @@ class EndpointHandler:
         """
 
         # Get inputs
-        encrypted_inputs = data.pop("encrypted_inputs", data)
+        encrypted_inputs = from_json(data.pop("encrypted_inputs", data))
 
         # Get keys
-        evaluation_keys = data.pop("evaluation_keys", data)
+        evaluation_keys = from_json(data.pop("evaluation_keys", data))
 
         # Run CML prediction
         encrypted_prediction = self.fhemodel_server.run(encrypted_inputs, evaluation_keys)