Commit
·
a240179
1
Parent(s):
42c394e
handler change1
Browse files- handler.py +7 -4
handler.py
CHANGED
|
@@ -28,12 +28,15 @@ class EndpointHandler:
|
|
| 28 |
self.model.set_generation_params(duration=duration)
|
| 29 |
outputs = self.model.generate(inputs)
|
| 30 |
|
| 31 |
-
# Save each generated audio file with loudness normalization
|
| 32 |
output_files = []
|
| 33 |
for idx, one_wav in enumerate(outputs):
|
| 34 |
output_file_path = f"generated_audio_{idx}.wav"
|
| 35 |
audio_write(output_file_path, one_wav.cpu(), self.model.sample_rate, strategy="loudness", loudness_compressor=True)
|
| 36 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
|
| 38 |
-
return {"generated_audio_files": output_files}
|
| 39 |
-
|
|
|
|
| 28 |
self.model.set_generation_params(duration=duration)
|
| 29 |
outputs = self.model.generate(inputs)
|
| 30 |
|
| 31 |
+
# Save each generated audio file with loudness normalization and encode in base64
|
| 32 |
output_files = []
|
| 33 |
for idx, one_wav in enumerate(outputs):
|
| 34 |
output_file_path = f"generated_audio_{idx}.wav"
|
| 35 |
audio_write(output_file_path, one_wav.cpu(), self.model.sample_rate, strategy="loudness", loudness_compressor=True)
|
| 36 |
+
|
| 37 |
+
# Read the file and encode it in base64
|
| 38 |
+
with open(output_file_path, "rb") as audio_file:
|
| 39 |
+
encoded_string = base64.b64encode(audio_file.read()).decode('utf-8')
|
| 40 |
+
output_files.append(encoded_string)
|
| 41 |
|
| 42 |
+
return {"generated_audio_files": output_files}
|
|
|