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