Update app.py

app.py

@@ -41,8 +41,14 @@ def style_transfer(prompt, negative_prompt, audio_input):
     spec = spectro_from_wav(audio_input)
     print(spec)
     # Open the image
-    im = image_from_spectrogram(spec)
-   
+    im = Image.open(spec)
+    
+    # Convert the image to a NumPy array
+    im = np.array(im)
+    # Open the image
+    im = image_from_spectrogram(im)
+    # Convert the image to a NumPy array
+    
     new_spectro = pipe2(prompt=prompt, image=im, strength=0.5, guidance_scale=7).images
     wav = wav_bytes_from_spectrogram_image(new_spectro[0])
     with open("output.wav", "wb") as f: