Update app.py

app.py

@@ -6,11 +6,13 @@ import gradio as gr
 
 # Load the StableAudio pipeline model
 pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+
 
 # Define the function to generate the sound based on a text prompt
 @spaces.GPU
 def generate_sound(prompt, negative_prompt, seed, inference_steps, duration, waveforms):
+    pipe = pipe.to("cuda")
+    
     # Set the seed for reproducibility
     generator = torch.Generator("cuda").manual_seed(seed)