qhub-blip-image-captioning-finetuned / api_client_sample.py
sguna's picture
Upload api_client_sample.py
b13e92e verified
raw
history blame
876 Bytes
import base64
import io
import requests
from PIL import Image
API_URL = "https://xt41z6d0qly7bg8w.us-east-1.aws.endpoints.huggingface.cloud"
headers = {
"Accept": "application/json",
"Authorization": "Bearer hf_###",
"Content-Type": "application/json"
}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.json()
image_path = 'car_clock.jpg' # Replace with the path to your image file
# Open the image and convert it to bytes
with Image.open(image_path) as img:
img_bytes = io.BytesIO()
img.save(img_bytes, format='JPEG')
image_bytes = img_bytes.getvalue()
image_base64 = base64.b64encode(image_bytes).decode('utf-8')
output = query({
"inputs": {
"text": "",
"image": image_base64
},
"parameters": {
"max_new_tokens": 150
}
})
print(output)