Update app.py

app.py

@@ -34,14 +34,10 @@ 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,
-        )
+        pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-audio-open-1.0",torch_dtype=DTYPE,trust_remote_code=True,revision="main")
 
-        pipe.to(DEVICE)
-        pipe.set_progress_bar_config(disable=True)
+        pipe.enable_attention_slicing()
+        pipe.enable_model_cpu_offload()
 
         print("Pipeline loaded on", DEVICE)