Update app.py

app.py

@@ -16,7 +16,7 @@ def predict(prompt, duration):
         width_duration=512
     else :
         width_duration = 512 + ((int(duration)-5) * 128)
-    spec = pipe(prompt, height=256, width=512).images[0]
+    spec = pipe(prompt, height=512, width=width_duration).images[0]
     print(spec)
     wav = wav_bytes_from_spectrogram_image(spec)
     with open("output.wav", "wb") as f: