| from handler import EndpointHandler | |
| import soundfile as sf | |
| import numpy as np | |
| # init handler | |
| my_handler = EndpointHandler(path=".") | |
| # prepare sample payload | |
| payload = {"inputs": "Lowfi hiphop with deep bass"} | |
| # test the handler | |
| pred=my_handler(payload) | |
| audio_data = np.array(pred["audio_data"], dtype=np.float32) | |
| sampling_rate = pred["sampling_rate"] | |
| # Write the audio data to a WAV file | |
| output_file_path = "output.wav" # Specify the file path | |
| sf.write(output_file_path, audio_data, sampling_rate) | |
| print("Audio file saved successfully.") | |