| from handler import EndpointHandler | |
| import base64 | |
| # init handler | |
| my_handler = EndpointHandler(path=".") | |
| # prepare sample payload | |
| text_payload = {"text": "I am quite excited how this will turn out"} | |
| audio_payload = {"audio": base64.b64encode( | |
| open("/home/arthur/data/musicdb/split_demucses/474499231_456751864 --__-- Breakbot - Programme_mp3_drums/chunk_11.wav", 'rb').read())} | |
| # test the handler | |
| # text_pred = my_handler(text_payload) | |
| audio_pred = my_handler(audio_payload) | |
| # show results | |
| # print("text_pred", text_pred) | |
| print("audio_pred", audio_pred) | |