fthor's picture
Added handler.py and requirements.txt
4b86165
raw
history blame contribute delete
No virus
1.56 kB
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}]