Update app.py

app.py

@@ -25,13 +25,12 @@ def predict(text):
     
     with torch.autocast("cuda"):
         outputs = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=512)
-        data = outputs[0].cpu().numpy().astype(np.float16)
 
-        video_path = gr.make_waveform((32000, data.ravel()), bars_color=random.choice(COLORS), bar_count=75)
+        video_path = gr.make_waveform((32000, outputs[0].cpu().numpy().astype(np.float16).ravel()), bars_color=random.choice(COLORS), bar_count=75)
         command = f'ffmpeg -y -i {video_path} -vf "scale=250:150" result.mp4'
         subprocess.run(command, shell=True) 
 
-        return (32000, data), "result.mp4"
+        return (32000, outputs[0][0].cpu().numpy().astype(np.float16)), "result.mp4"
 
 title = "MusicGen"