from handler import EndpointHandler | |
from PIL import Image | |
import requests, json | |
# init handler | |
my_handler = EndpointHandler(use_cuda=False, test_mode=True) | |
# API_URL = "https://oncm9ojdmjwesag2.eu-west-1.aws.endpoints.huggingface.cloud" | |
# headers = { | |
# "Authorization": "Bearer MY_API_TOKEN", | |
# "Content-Type": "image/jpg" | |
# } | |
# def query(filename): | |
# with open(filename, "rb") as f: | |
# data = f.read() | |
# response = requests.request("POST", API_URL, headers=headers, data=data) | |
# return json.loads(response.content.decode("utf-8")) | |
# output = query("food.jpg") | |
# prepare sample payload | |
image_path = '/Users/francois/Documents/dev/Maurice/maurice/test_602.jpg' | |
with open(image_path, 'rb') as f: | |
img = f.read() | |
single_image = { | |
'inputs': [ | |
img | |
], | |
'prompt': 'Describe the image' | |
} | |
multiple_images = { | |
'inputs': [ | |
img, img, img | |
], | |
'prompt': 'Describe the image' | |
} | |
# test the handler | |
print(my_handler(single_image)) | |
print(my_handler(multiple_images)) | |
# non_holiday_payload = {"inputs": "I am quite excited how this will turn out", "date": "2022-08-08"} | |
# holiday_payload = {"inputs": "Today is a though day", "date": "2022-07-04"} | |
# | |
# | |
# | |
# # test the handler | |
# non_holiday_pred=my_handler(non_holiday_payload) | |
# holiday_payload=my_handler(holiday_payload) | |
# | |
# # show results | |
# print("non_holiday_pred", non_holiday_pred) | |
# print("holiday_payload", holiday_payload) | |
# | |
# # non_holiday_pred [{'label': 'joy', 'score': 0.9985942244529724}] | |
# # holiday_payload [{'label': 'happy', 'score': 1}] | |