Add health check endpoint

stackhpc · Nov 12, 2024 · 2cc0484 · 2cc0484
1 parent 8a38a86
commit 2cc0484
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/web-apps/flux-image-gen/api_server.py b/web-apps/flux-image-gen/api_server.py
@@ -31,6 +31,9 @@ class ImageGenInput(BaseModel):
     prompt: str
     add_sampling_metadata: bool
 
+@app.get("/")
+def health_check():
+    return "Server is running"
 
 @app.get("/model")
 async def get_model():