Update app.py

app.py

@@ -34,15 +34,21 @@ def load_pipeline():
     if pipe is None:
         print("Loading Stable Audio Open pipeline...")
 
-        pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-audio-open-1.0",torch_dtype=DTYPE,trust_remote_code=True,revision="main")
+        pipe = DiffusionPipeline.from_pretrained(
+            "stabilityai/stable-audio-open-1.0",
+            torch_dtype=DTYPE,
+            trust_remote_code=True,
+            revision="main",
+            low_cpu_mem_usage=False,
+        )
 
         pipe.enable_attention_slicing()
         pipe.to(DEVICE)
 
-
         print("Pipeline loaded on", DEVICE)
 
 
+
 @app.on_event("startup")
 def startup():
     load_pipeline()