Upload handler.py

handler.py

@@ -8,6 +8,14 @@ def from_json(python_object):
         return bytes(python_object["__value__"])
 
 
+def to_json(python_object):
+    if isinstance(python_object, bytes):
+        return {"__class__": "bytes", "__value__": list(python_object)}
+    # if isinstance(python_object, bytes):
+    #     return list(python_object)
+    raise TypeError(repr(python_object) + " is not JSON serializable")
+
+
 class EndpointHandler:
     def __init__(self, path="benoit.model"):
 
@@ -32,4 +40,4 @@ class EndpointHandler:
         # Run CML prediction
         encrypted_prediction = self.fhemodel_server.run(encrypted_inputs, evaluation_keys)
 
-        return encrypted_prediction
+        return to_json(encrypted_prediction)