chore: instance in europe

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://yw1dgyuig6ff5pft.us-east-1.aws.endpoints.huggingface.cloud"
+API_URL = "https://h0cvbig1fkmf57eb.eu-west-1.aws.endpoints.huggingface.cloud"
 headers = {
     "Authorization": "Bearer " + os.environ.get("HF_TOKEN"),
     "Content-Type": "application/json",
@@ -86,8 +85,10 @@ for i in range(nb_samples):
 
     # Run the inference on HF servers
     duration -= time.time()
+    duration_inference = -time.time()
     encrypted_prediction = query(payload)
     duration += time.time()
+    duration_inference += time.time()
 
     encrypted_prediction = from_json(encrypted_prediction)
 
@@ -96,7 +97,9 @@ for i in range(nb_samples):
     prediction = np.argmax(prediction_proba)
 
     if verbose or True:
-        print(f"for {i}-th input, {prediction=} with expected {Y_test[i]}")
+        print(
+            f"for {i}-th input, {prediction=} with expected {Y_test[i]} in {duration_inference} seconds"
+        )
 
     # Measure accuracy
     nb_good += Y_test[i] == prediction