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}