Update ocr_server.py

magic_pdf/dict2md/ocr_server.py

@@ -26,7 +26,7 @@ REVISION = 'v1.0.4'
 BOX_TAG_PATTERN = r"<box>([\s\S]*?)</box>"
 PUNCTUATION = "！？。＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』【】〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏."
 
-logger.add("../../parse.log", rotation="10 MB", level="INFO",
+logger.add("parse.log", rotation="10 MB", level="INFO",
            format="{time} {level} {message}", encoding='utf-8', enqueue=True)
 
 def _get_args():
@@ -226,6 +226,10 @@ class Item(BaseModel):
     image_path: str
     text: str
 
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy"}
+
 
 @app.post("/predict")
 async def predict(item: Item):