Update app.py

app.py

@@ -23,7 +23,7 @@ def get_spectro(prompt):
     image = pipe(prompt).images[0] 
     return image
 
-def wav_bytes_from_spectrogram_image(image):
+def wav_bytes_from_spectrogram_image(image: Image.Image) -> T.Tuple[io.BytesIO, float]:
     """
     Reconstruct a WAV audio clip from a spectrogram image. Also returns the duration in seconds.
     """
@@ -71,7 +71,7 @@ def wav_bytes_from_spectrogram_image(image):
     return wav_bytes
 
 def spectrogram_from_image(
-    image, max_volume: float = 50, power_for_image: float = 0.25
+    image: Image.Image, max_volume: float = 50, power_for_image: float = 0.25
 ) -> np.ndarray:
     """
     Compute a spectrogram magnitude array from a spectrogram image.