from handler import EndpointHandler

# init handler
my_handler = EndpointHandler(path="pbotsaris/musicgen-small")

non_holiday_payload = {"inputs": "I am quite excited how this will turn out", "duration": 2}
# prepare sample payload

# test the handler
non_holiday_pred=my_handler(non_holiday_payload)

# show results
print("done")
print(non_holiday_pred)

# non_holiday_pred [{'label': 'joy', 'score': 0.9985942244529724}]
# holiday_payload [{'label': 'happy', 'score': 1}]