Update app.py

app.py

@@ -19,9 +19,12 @@ def generate_music(prompt, duration):
     # Prepare inputs
     inputs = processor(text=[prompt], return_tensors="pt").to(device)
 
-    # ✅ Use exact audio length parameter instead of token approximation
+    # Approximate duration scaling
+    approx_tokens = int(256 * (duration / 8))
+    approx_tokens = min(approx_tokens, 2048)
+
     with torch.no_grad():
-        audio = model.generate(**inputs, do_sample=True, guidance_scale=3, audio_length_in_s=int(duration))
+        audio = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=approx_tokens)
 
     sr = model.config.audio_encoder.sampling_rate
     audio_arr = audio[0, 0].cpu().numpy()
@@ -30,10 +33,10 @@ def generate_music(prompt, duration):
     tmp = tempfile.NamedTemporaryFile(suffix=".wav", delete=False)
     wavfile.write(tmp.name, rate=sr, data=audio_arr)
 
-    return tmp.name, f"✅ Generated {duration}s of audio!"
+    return tmp.name, f"✅ Generated {duration}s (approx) of audio!"
 
 with gr.Blocks(title="MusicGen 🎶") as demo:
-    gr.Markdown("# 🎵 MusicGen — Text-to-Music Generator (Accurate Duration 40s Version)")
+    gr.Markdown("# 🎵 MusicGen — Text-to-Music Generator (Stable 40s Version)")
     with gr.Row():
         prompt = gr.Textbox(label="🎼 Describe your music", placeholder="e.g. dreamy lo-fi with soft piano")
         duration = gr.Slider(4, 40, value=15, step=1, label="Duration (seconds)")