Update app.py

app.py

@@ -25,12 +25,7 @@ from TTS.api import TTS
 # Diffusers for sound design generation
 from diffusers import DiffusionPipeline, AudioLDMPipeline
 import diffusers
-
-# Monkey-patch: Create a patched pipeline class so that any reference to AudioLDM2Pipeline is resolved correctly.
-class PatchedAudioLDM2Pipeline(AudioLDMPipeline):
-    pass
-
-setattr(diffusers, "AudioLDM2Pipeline", PatchedAudioLDM2Pipeline)
+from packaging import version
 
 # ---------------------------------------------------------------------
 # Setup Logging and Environment Variables
@@ -107,11 +102,16 @@ 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 using the patched pipeline class.
+    otherwise, it loads and caches the pipeline.
+    
+    NOTE: AudioLDM2Pipeline is available only in diffusers>=0.21.0.
+    Since your requirements fix diffusers==0.20.2, this function will raise an error.
     """
+    if version.parse(diffusers.__version__) < version.parse("0.21.0"):
+        raise ValueError("AudioLDM2 requires diffusers>=0.21.0. Please upgrade your diffusers package.")
     if model_name in SOUND_DESIGN_PIPELINES:
         return SOUND_DESIGN_PIPELINES[model_name]
-    pipe = DiffusionPipeline.from_pretrained(model_name, pipeline_class=PatchedAudioLDM2Pipeline, use_auth_token=token)
+    pipe = DiffusionPipeline.from_pretrained(model_name, pipeline_class=AudioLDMPipeline, use_auth_token=token)
     SOUND_DESIGN_PIPELINES[model_name] = pipe
     return pipe