Update spectro.py

spectro.py

@@ -65,10 +65,11 @@ def spectrogram_from_image(
 ) -> np.ndarray:
     """
     Compute a spectrogram magnitude array from a spectrogram image.
+
     TODO(hayk): Add image_from_spectrogram and call this out as the reverse.
     """
     # Convert to a numpy array of floats
-    data = np.array(image.convert('RGBA')).astype(np.float32)
+    data = np.array(image).astype(np.float32)
 
     # Flip Y take a single channel
     data = data[::-1, :, 0]
@@ -85,7 +86,6 @@ def spectrogram_from_image(
     return data
 
 
-
 def spectrogram_from_waveform(
     waveform: np.ndarray,
     sample_rate: int,