Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 
+from PIL import Image
 from spectro import wav_bytes_from_spectrogram_image
 
 from diffusers import StableDiffusionPipeline
@@ -38,7 +39,12 @@ def classic(prompt, negative_prompt, duration):
 def style_transfer(prompt, negative_prompt, audio_input):
     spec = spectro_from_wav(audio_input)
     print(spec)
-    new_spectro = pipe2(prompt=prompt, image=spec, strength=0.5, guidance_scale=7).images
+    # Open the image
+    im = Image.open('image.png')
+    
+    # Convert the image to a NumPy array
+    image = np.array(im)
+    new_spectro = pipe2(prompt=prompt, image=image, strength=0.5, guidance_scale=7).images
     wav = wav_bytes_from_spectrogram_image(new_spectro[0])
     with open("output.wav", "wb") as f:
         f.write(wav[0].getbuffer())