test.py

# from handler import EndpointHandler
#
# # init handler
# my_handler = EndpointHandler(path=".")
#
# # prepare sample payload
# 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}]
from handler import EndpointHandler

handler = EndpointHandler(path=".")

# Example input data
test_data = {
    "inputs": "John Doe visited New York last week."
}

# Invoke the handler
result = handler(test_data)
print(result)