Update app.py

app.py

@@ -100,11 +100,13 @@ def get_tts_model(model_name: str = "tts_models/en/ljspeech/tacotron2-DDC"):
 def get_sound_design_pipeline(model_name: str, token: str):
     """
     Returns a cached DiffusionPipeline for sound design if available;
-    otherwise, it loads and caches the pipeline.
+    otherwise, it loads and caches the pipeline using the correct pipeline class.
     """
     if model_name in SOUND_DESIGN_PIPELINES:
         return SOUND_DESIGN_PIPELINES[model_name]
-    pipe = DiffusionPipeline.from_pretrained(model_name, use_auth_token=token)
+    # Import the correct pipeline class from diffusers
+    from diffusers import AudioLDMPipeline
+    pipe = DiffusionPipeline.from_pretrained(model_name, pipeline_class=AudioLDMPipeline, use_auth_token=token)
     SOUND_DESIGN_PIPELINES[model_name] = pipe
     return pipe
 
@@ -219,7 +221,7 @@ def generate_music(prompt: str, audio_length: int):
 @spaces.GPU(duration=200)
 def generate_sound_design(prompt: str):
     """
-    Generates a sound design audio file based on the provided prompt using Audioldm2.
+    Generates a sound design audio file based on the provided prompt using Audioldm.
     Returns the file path to the generated .wav file.
     """
     try: