health check

DARPA-ASKEM · Sep 11, 2024 · c76afcd · c76afcd
1 parent 8c4789e
commit c76afcd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/document_intelligence/fast_latex/run.py b/document_intelligence/fast_latex/run.py
@@ -101,6 +101,10 @@ async def process_and_predict(file: UploadFile = File(...), isolated_only: bool
 
     return JSONResponse(content=latex_results)
 
+@app.get("/health")
+async def health_check():
+    return JSONResponse(content={"status": "OK"}, status_code=200)
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8002)
diff --git a/document_intelligence/nougat/run.py b/document_intelligence/nougat/run.py
@@ -181,6 +181,10 @@ async def process_pdf(file: UploadFile = File(...)):
 
     return JSONResponse(content=latex_dict)
 
+@app.get("/health")
+async def health_check():
+    return JSONResponse(content={"status": "OK"}, status_code=200)
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)