Update handler.py

handler.py

@@ -22,6 +22,10 @@ class EndpointHandler:
         duration = inputs.get("duration", 10)
         parameters = data.get("parameters", {})
 
+        # Validate the prompt
+        if not prompt:
+            return {"error": "No prompt provided."}
+
         # Preprocess the prompt
         input_ids = self.processor(
             text=[prompt],
@@ -48,4 +52,4 @@ class EndpointHandler:
                 "generated_audio": audio_list,
                 "sample_rate": self.sampling_rate,
             }
-        ]
+        ]