adding health check to endpoint

Dockerfile

@@ -12,4 +12,7 @@ COPY --chown=user requirements.txt .
 RUN pip install  --user -r requirements.txt
 
 COPY --chown=user . .
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl --fail http://localhost:8080/test || exit 1
+
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -64,6 +64,11 @@ def infer_shakkala(request: ShakkalaRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 '''
+
+@app.get("/test")
+def health_check():
+    return {"status": "ok"}
+
 @app.post("/catt")
 def infer_catt(request: CattRequest):
     try: