Update README.md
Browse files
README.md
CHANGED
|
@@ -14,4 +14,54 @@ To use deploy this model a an Inference Endpoint you have to select `Custom` as
|
|
| 14 |
|
| 15 |
The repository contains a requirements.txt to download the einops, timm and pillow library.
|
| 16 |
|
|
|
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
The repository contains a requirements.txt to download the einops, timm and pillow library.
|
| 16 |
|
| 17 |
+
## Call to endpoint example
|
| 18 |
|
| 19 |
+
``` python
|
| 20 |
+
import json
|
| 21 |
+
from typing import List
|
| 22 |
+
import requests as r
|
| 23 |
+
import base64
|
| 24 |
+
|
| 25 |
+
ENDPOINT_URL = "endpoint_url"
|
| 26 |
+
HF_TOKEN = "token_key"
|
| 27 |
+
|
| 28 |
+
def predict(path_to_image: str = None, text : str = None):
|
| 29 |
+
with open(path_to_image, "rb") as i:
|
| 30 |
+
b64 = base64.b64encode(i.read())
|
| 31 |
+
|
| 32 |
+
payload = {"inputs":
|
| 33 |
+
{
|
| 34 |
+
"image": b64.decode("utf-8"),
|
| 35 |
+
"text": text
|
| 36 |
+
}
|
| 37 |
+
}
|
| 38 |
+
|
| 39 |
+
response = r.post(
|
| 40 |
+
ENDPOINT_URL, headers={"Authorization": f"Bearer {HF_TOKEN}"}, json=payload
|
| 41 |
+
)
|
| 42 |
+
return response.json()
|
| 43 |
+
|
| 44 |
+
|
| 45 |
+
prediction = predict(
|
| 46 |
+
path_to_image="image/accidentdevoiture.webp", text="An image of a cat and a remote control"
|
| 47 |
+
)
|
| 48 |
+
|
| 49 |
+
print(json.dumps(prediction, indent=2))
|
| 50 |
+
```
|
| 51 |
+
|
| 52 |
+
## Expected result
|
| 53 |
+
|
| 54 |
+
``` json
|
| 55 |
+
{
|
| 56 |
+
"text_embedding": [-0.009289545938372612,
|
| 57 |
+
-0.03686045855283737,
|
| 58 |
+
...
|
| 59 |
+
0.038627129048109055,
|
| 60 |
+
-0.01346363127231597]
|
| 61 |
+
"image_embedding": [-0.009289545938372612,
|
| 62 |
+
-0.03686045855283737,
|
| 63 |
+
...
|
| 64 |
+
0.038627129048109055,
|
| 65 |
+
-0.01346363127231597]
|
| 66 |
+
}
|
| 67 |
+
```
|