chore: using a database for keys

handler.py

@@ -49,17 +49,21 @@ class EndpointHandler:
 
             return uid
 
-        assert method == "inference":
+        elif method == "inference":
 
-        uid = data.pop("uid", data)
+            uid = data.pop("uid", data)
 
-        # Get inputs
-        encrypted_inputs = from_json(data.pop("encrypted_inputs", data))
+            # Get inputs
+            encrypted_inputs = from_json(data.pop("encrypted_inputs", data))
 
-        # Find key in the database
-        evaluation_keys = self.key_database[uid]
+            # Find key in the database
+            evaluation_keys = self.key_database[uid]
 
-        # Run CML prediction
-        encrypted_prediction = self.fhemodel_server.run(encrypted_inputs, evaluation_keys)
+            # Run CML prediction
+            encrypted_prediction = self.fhemodel_server.run(encrypted_inputs, evaluation_keys)
 
-        return to_json(encrypted_prediction)
+            return to_json(encrypted_prediction)
+
+        else:
+
+            return