Update modules/audio_ldm.py

modules/audio_ldm.py

@@ -11,8 +11,9 @@ class audio_generation:
     
     def infer(self, prompt, num_length=10, negative="low quality, background sound, average quality, horrible quality"):  
         prompt = str(prompt)
-        num_length = num_length.replace('"', '').replace("'", '')
-        num_length = int(num_length)
+        if isinstance(num_length, int):
+            num_length = num_length.replace('"', '').replace("'", '')
+            num_length = int(num_length)
         audio = self.pipe(prompt, num_inference_steps=20, audio_length_in_s=num_length, negative_prompt=negative).audios[0]
         num = random.randint(1, 1000)
         audio_file = f"audio{num}.wav"