chore: adding a test with DL

play_with_endpoint.py

@@ -20,8 +20,7 @@ def from_json(python_object):
         return bytes(python_object["__value__"])
 
 
-# TODO: put the right link `API_URL` for your entry point
-API_URL = "https://XXXXXXX.us-east-1.aws.endpoints.huggingface.cloud"
+API_URL = "https://ml79quhm2orc8dbo.eu-west-1.aws.endpoints.huggingface.cloud"
 headers = {
     "Authorization": "Bearer " + os.environ.get("HF_TOKEN"),
     "Content-Type": "application/json",
@@ -39,12 +38,13 @@ def query(payload):
 
 path_to_model = Path("compiled_model")
 
-# BEGIN: replace this part with your privacy-preserving application
-from sklearn.datasets import make_classification
+# DL in FHE
+from sklearn.datasets import load_digits
 from sklearn.model_selection import train_test_split
 
-x, y = make_classification(n_samples=1000, class_sep=2, n_features=30, random_state=42)
-_, X_test, _, Y_test = train_test_split(x, y, test_size=0.2, random_state=42)
+X, y = load_digits(return_X_y=True)
+X = np.expand_dims(X.reshape((-1, 8, 8)), 1)
+_, X_test, _, Y_test = train_test_split(X, y, test_size=0.25, shuffle=True, random_state=42)
 
 # Recover parameters for client side
 fhemodel_client = FHEModelClient(path_to_model)
@@ -72,7 +72,7 @@ while not is_finished:
         is_finished = True
 
     print(
-        f"Sending {i}-th piece of the key (remaining size is {sys.getsizeof(evaluation_keys_remaining)})"
+        f"Sending {i}-th piece of the key (remaining size is {sys.getsizeof(evaluation_keys_remaining) / 1024:.2f} kbytes)"
     )
     i += 1
 
@@ -99,7 +99,7 @@ while not is_finished:
 
 # Test the handler
 nb_good = 0
-nb_samples = len(X_test)
+nb_samples = 10  # len(X_test)
 verbose = True
 time_start = time.time()
 duration = 0
@@ -119,7 +119,7 @@ for i in range(nb_samples):
     }
 
     if is_first:
-        print(f"Size of the payload: {sys.getsizeof(payload) / 1024} kilobytes")
+        print(f"Size of the payload: {sys.getsizeof(payload) / 1024:.2f} kilobytes")
         is_first = False
 
     # Run the inference on HF servers