Yan
removed useless character from the file, removed specific endpoint url from endpoint tester
609d64d
| import json | |
| from typing import List | |
| import requests as r | |
| import base64 | |
| from PIL import Image | |
| from io import BytesIO | |
| ENDPOINT_URL = "https://XXXXXX" # your endpoint url | |
| HF_TOKEN = "#######################" # your huggingface token `hf_xxx` | |
| # helper image utils | |
| def encode_image(image_path): | |
| with open(image_path, "rb") as i: | |
| b64 = base64.b64encode(i.read()) | |
| return b64.decode("utf-8") | |
| def predict(image): | |
| image = encode_image(image) | |
| # prepare sample payload | |
| payload = {"image": image, "inputs": ""} | |
| # headers | |
| headers = { | |
| "Authorization": f"Bearer {HF_TOKEN}", | |
| "Content-Type": "application/json", | |
| } | |
| response = r.post(ENDPOINT_URL, headers=headers, json=payload) | |
| if response.status_code != 200: | |
| print(response.text) | |
| raise Exception("Prediction failed") | |
| return response.json() | |
| prediction = predict( | |
| image = "test.jpeg" | |
| ) | |
| print(prediction[1]) | |
| print(prediction[2]) | |