Update app.py

app.py

@@ -95,7 +95,7 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
         sigma_min=0.3,
         sigma_max=500,
         sampler_type="dpmpp-3m-sde",
-        device=device,
+        device="cuda",
     )
     # [B, C, N] -> [C, B*N] -> [N, C] for Gradio
     audio = rearrange(audio, "b c n -> c (b n)")           # (C, T)
@@ -226,9 +226,6 @@ interface = gr.Interface(
     """
 )
 
-# Pre-load the model to avoid multiprocessing issues
-model, model_config = load_model()
-
 # Launch the Interface
 if __name__ == "__main__":
     interface.launch()