Update handler.py

handler.py

@@ -29,13 +29,9 @@ class EndpointHandler:
         prompt_input_ids = self.tokenizer(prompt, return_tensors="pt").input_ids.to(self.device)
         
         set_seed(42)
-        try:
-            generation = self.model.generate(input_ids=input_ids, prompt_input_ids=prompt_input_ids)
-            audio_arr = generation.cpu().numpy().squeeze()
-            sf.write("parler_tts_out.wav", audio_arr, self.model.config.sampling_rate)
-            with open("parler_tts_out.wav", "rb") as f:
-                return base64.b64encode(f.read()).decode()
-        
-        except Exception as e:
-            logger.error(str(e))
-            return {"error": str(e)}
+        generation = self.model.generate(input_ids=input_ids, prompt_input_ids=prompt_input_ids)
+        audio_arr = generation.cpu().numpy().squeeze()
+        sf.write("parler_tts_out.wav", audio_arr, self.model.config.sampling_rate)
+        with open("parler_tts_out.wav", "rb") as f:
+            return base64.b64encode(f.read()).decode()
+