import math | |
from handler import EndpointHandler | |
from scipy.io.wavfile import write | |
# init handler | |
my_handler = EndpointHandler(path=".") | |
generation_params = { | |
"duration": 12, | |
} | |
# prepare sample payload | |
payload = { | |
"prompt": "rock, rock and rock", | |
"generation_params": generation_params, | |
"audio_window": 2, | |
"chunk_size": 4, | |
} | |
# test the handler | |
test = my_handler(payload) | |
write("test.wav", 32000, test["generated_audio"]) | |