Convert to RGB

2022-12-16 16:26:46 -06:00 · 2022-12-16 16:26:46 -06:00 · 1c5caf77ac
parent 9ab8c0f105
commit 1c5caf77ac
1 changed files with 3 additions and 1 deletions
--- a/riffusion/audio.py
+++ b/riffusion/audio.py
@ -105,10 +105,12 @@ def image_from_spectrogram(

    # Flip Y
    image = image.transpose(Image.FLIP_TOP_BOTTOM)
+    
+    # Convert to RGB
+    image = image.convert("RGB")

    return image

-
 def spectrogram_from_waveform(
    waveform: np.ndarray,
    sample_rate: int,