test_handler.py

from handler import EndpointHandler
import json
from typing import List
import requests as r
import base64
import requests as r
import base64
from PIL import Image
from io import BytesIO

ENDPOINT_URL = ""
HF_TOKEN = ""
def decode_base64_image(image_string):
  base64_image = base64.b64decode(image_string)
  buffer = BytesIO(base64_image)
  return  Image.open(buffer)

# init handler
my_handler = EndpointHandler(path=".")

# prepare sample payload
path_to_image = "test_images/lal.jpg"
with open(path_to_image, "rb") as i:
    b64 = base64.b64encode(i.read())

payload = {"inputs": {"image": b64.decode("utf-8"), "class_text": "shirt", "prompt": "wedding shirt"}}

# test the handler
results=my_handler(payload)

# show results
# print("non_holiday_pred", non_holiday_pred)
# print("holiday_payload", holiday_payload)
decode_base64_image(results["image"]).save("test_results.jpg")
# def predict(path_to_image: str = None, candiates: List[str] = None):
#     with open(path_to_image, "rb") as i:
#         b64 = base64.b64encode(i.read())

#     payload = {"inputs": {"image": b64.decode("utf-8"), "candiates": candiates}}
#     response = r.post(
#         ENDPOINT_URL, headers={"Authorization": f"Bearer {HF_TOKEN}"}, json=payload
#     )
#     return response.json()


# prediction = predict(
#     path_to_image="palace.jpg", candiates=["sea", "palace", "car", "ship"]
# )