Update app.py

app.py

@@ -41,12 +41,6 @@ def generate_audio():
         # Generate the audio using StableAudioPipeline
         generator = torch.Generator(device)
         generator.manual_seed(seed) 
-
-        latent_noise = torch.randn(
-            (1, 2, pipe.vae.latent_dim, pipe.vae.latent_dim),  # Use dimensions compatible with the VAE
-            device=device,
-            generator=generator
-        )
         
         audio_output = pipe(
             prompt=prompt,
@@ -55,8 +49,7 @@ def generate_audio():
             guidance_scale=10.0,
             audio_end_in_s=1,
             num_waveforms_per_prompt=1,
-            generator=generator,
-            latents = latent_noise
+            generator=generator
         ).audios
         
         # Convert audio to BytesIO in memory