Mirrors
/
riffusion-inference
mirror of https://github.com/riffusion/riffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

save JPEG instead of PNG

This commit is contained in:
Hayk Martiros 2022-11-26 19:24:50 +00:00
parent cd4c00b12e
commit 7f27705f81
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
riffusion/server.py
View File

@ -139,7 +139,7 @@ def run_inference():
mp3_bytes = mp3_bytes_from_wav_bytes(wav_bytes)
# Compute the output as base64 encoded strings
image_bytes = image_bytes_from_image(image, mode="PNG")
image_bytes = image_bytes_from_image(image, mode="JPEG")
output = InferenceOutput(image=base64_encode(image_bytes), audio=base64_encode(mp3_bytes))
# Log the total time