Change width duration to 15

app.py

@@ -39,7 +39,12 @@ def get_bg_image(prompt):
     return "img.png"
 
 def get_music(prompt):
-    spec = pipe2(prompt, height=512, width=512).images[0]
+    duration = 15
+    if duration == 5:
+        width_duration=512
+    else :
+        width_duration = 512 + ((int(duration)-5) * 128)
+    spec = pipe2(prompt, height=512, width=width_duration).images[0]
     print(spec)
     wav = wav_bytes_from_spectrogram_image(spec)
     with open("output.wav", "wb") as f: