test.py

from handler import EndpointHandler

# init handler
my_handler = EndpointHandler(path=".")

# prepare sample payload
payload = {"input": "Lowfi hiphop with deep bass"}

# test the handler
pred=my_handler(payload)


with open("generated_audio.wav", "wb") as f:
    f.write(pred)