arguments to handleer

handler.py

@@ -12,14 +12,23 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 
 class EndpointHandler:
- def __call__(self, data: Dict[str, bytes]) -> Dict[str, List[Any]]:
- """
- Args:
- data (:obj:):
- includes the deserialized image file as PIL.Image
- """
- # process input
- image = data.pop("inputs", data)
+ 
+ def __init__(self, path=""):
+ self.pytesseract_installed = False
+ try:
+ import pytesseract 
+ self.pytesseract_installed = True
+ except ImportError:
+ print("Pytesseract not installed, will not use OCR")
+ 
+ def __call__(self, data: Dict[str, bytes]) -> Dict[str, List[Any]]:
+ """
+ Args:
+ data (:obj:):
+ includes the deserialized image file as PIL.Image
+ """
+ # process input
+ image = data.pop("inputs", data)
 
- result = pytesseract.image_to_string(image)
- return {"predictions": result}
+ result = pytesseract.image_to_string(image)
+ return {"predictions": result}