| 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}] | |