chore: using a database for keys

play_with_endpoint.py

@@ -32,12 +32,10 @@ headers = {
 
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
-    print(f"{response=}")
 
     if "error" in response:
         assert False, f"Got an error: {response=}"
 
-    print(f"{response.json()=}")
     return response.json()
 
 
@@ -59,7 +57,7 @@ payload = {
     "method": "save_key",
 }
 
-uid = query(payload)
+uid = query(payload)["uid"]
 print(f"Storing the key in the database under {uid=}")
 
 # Test the handler
@@ -82,6 +80,9 @@ for i in range(nb_samples):
         "uid": uid,
     }
 
+    if verbose or True:
+        print(f"Size of the payload: {sys.getsizeof(payload) / 1024} kilobytes")
+
     # Run the inference on HF servers
     duration -= time.time()
     duration_inference = -time.time()