Update app.py

app.py

@@ -15,7 +15,7 @@ def predict(prompt, duration):
     if duration == 5:
         width_duration=512
     else :
-        width_duration = int(512 + ((duration-5) * 128))
+        width_duration = 512 + ((int(duration)-5) * 128)
     spec = pipe(prompt, height=512, width=width_duration).images[0]
     print(spec)
     wav = wav_bytes_from_spectrogram_image(spec)