chore: minor updates to be closer to the template

play_with_endpoint.py

@@ -73,9 +73,10 @@ print(f"Storing the key in the database under {uid=}")
 # Test the handler
 nb_good = 0
 nb_samples = len(X_test)
-verbose = False
+verbose = True
 time_start = time.time()
 duration = 0
+is_first = True
 
 for i in range(nb_samples):
 
@@ -90,8 +91,9 @@ for i in range(nb_samples):
         "uid": uid,
     }
 
-    if verbose or True:
+    if is_first:
         print(f"Size of the payload: {sys.getsizeof(payload) / 1024} kilobytes")
+        is_first = False
 
     # Run the inference on HF servers
     duration -= time.time()
@@ -106,7 +108,7 @@ for i in range(nb_samples):
     prediction_proba = fhemodel_client.deserialize_decrypt_dequantize(encrypted_prediction)[0]
     prediction = np.argmax(prediction_proba)
 
-    if verbose or True:
+    if verbose:
         print(
             f"for {i}-th input, {prediction=} with expected {Y_test[i]} in {duration_inference:.3f} seconds"
         )
@@ -115,6 +117,5 @@ for i in range(nb_samples):
     nb_good += Y_test[i] == prediction
 
 print(f"Accuracy on {nb_samples} samples is {nb_good * 1. / nb_samples}")
-print(f"Total time: {time.time() - time_start} seconds")
-print(f"Duration in inferences: {duration} seconds")
-print(f"Duration per inference: {duration / nb_samples} seconds")
+print(f"Total time: {time.time() - time_start:.3f} seconds")
+print(f"Duration per inference: {duration / nb_samples:.3f} seconds")