chore: debug

play_with_endpoint.py

@@ -86,15 +86,19 @@ for i in range(nb_samples):
         "evaluation_keys": evaluation_keys,
     }
 
+    print(f"{payload=}")
+
     # Run the inference on HF servers
     duration -= time.time()
+    print(f"Starting at {time.time()}")
     encrypted_prediction = query(payload)
+    print(f"Ending at {time.time()}")
     duration += time.time()
 
-    encrypted_prediction = from_json(encrypted_prediction)
-
     print(f"{encrypted_prediction=}")
 
+    encrypted_prediction = encrypted_prediction
+
     if is_first:
         is_first = False
         print(f"Size of the payload: {sys.getsizeof(payload)} bytes")