Fix logging

handler.py

@@ -23,25 +23,20 @@ class EndpointHandler:
         Return:
               A :obj:`list` | `dict`: will be serialized and returned
         """
-        checkpoints = ""
+        image = data.pop("inputs", data)
+        self.logger.info("image")
 
-        try:
-            image = data.pop("inputs", data)
-            checkpoints += "image reached\n"
+        # process image
+        inputs = self.processor(images=image, return_tensors="pt").to(self.device)
+        self.logger.info("inputs")
+        generated_ids = self.model.generate(**inputs, max_new_tokens=20)
+        self.logger.info("generated")
 
-            # process image
-            inputs = self.processor(images=image, return_tensors="pt").to(self.device)
-            checkpoints += "inputs reached\n"
-            generated_ids = self.model.generate(**inputs, max_new_tokens=20)
-            checkpoints += "generated\n"
-
-            # run prediction
-            generated_text: List[str] = self.processor.batch_decode(
-                generated_ids, skip_special_tokens=True
-            )
-            checkpoints += "decoded\n"
-        except Exception as e:
-            checkpoints += f"{e}\n"
+        # run prediction
+        generated_text = self.processor.batch_decode(
+            generated_ids, skip_special_tokens=True
+        )
+        self.logger.info("decoded")
 
         # decode output
-        return generated_text.append(checkpoints)
+        return generated_text