Update app.py

app.py

@@ -7,13 +7,12 @@ import spaces
 
 # 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)