Update handler

handler.py

@@ -25,9 +25,12 @@ class EndpointHandler():
         # Get inputs
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", None)
+        LOGGER.info("Data extracted.")
         # Preprocess
+        LOGGER.info(f"Start tokenizer: {inputs}")
         inputs_ids = self.tokenizer(inputs, return_tensors="pt").inputs_ids
         # Forward
+        LOGGER.info(f"Start generation.")
         if parameters is not None:
             outputs = self.model.generate(inputs_ids, **parameters)
         else:
@@ -35,4 +38,4 @@ class EndpointHandler():
         # Postprocess
         prediction = self.tokenizer.decode(outputs[0])
         LOGGER.info(f"Generated text: {prediction}")
-        return [{"generated_text": prediction}]
+        return {"generated_text": prediction}