djomo
/

MISTRALllux1000-7b-v5-GGUF

Inference Endpoints

Model card Files Files and versions Community

djomo commited on Feb 8

Commit

1616f3a

•

1 Parent(s): 7e929c6

Upload handler.py

Files changed (1) hide show

handler.py +6 -1

handler.py CHANGED Viewed

@@ -8,7 +8,12 @@ from paddleocr import PaddleOCR, draw_ocr
 class EndpointHandler():
     def __init__(self, path=""):
-        self.pipeline = PaddleOCR(lang="en",ocr_version="PP-OCRv4",show_log = False,use_gpu=True)
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:

 class EndpointHandler():
     def __init__(self, path=""):
+        self.pipeline = PaddleOCR(lang="en",ocr_version="PP-OCRv4",
+                                    show_log = False,use_gpu=True,
+                                    det_model_dir=path,
+                                    cls_model_dir=path,
+                                    rec_model_dir=path
+                                    )
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]: