Commit
·
58427fc
1
Parent(s):
9587da3
Update health endpoint to match requested format with hf_inference_api and fal-ai
Browse files- app/main.py +2 -10
app/main.py
CHANGED
|
@@ -190,19 +190,11 @@ async def shutdown_event():
|
|
| 190 |
# -------------------------------------------------
|
| 191 |
@app.get("/health")
|
| 192 |
def health_check(request: Request):
|
| 193 |
-
# Determine model type based on what's available
|
| 194 |
-
if CCO_AVAILABLE:
|
| 195 |
-
model_type = "cco-eccv16"
|
| 196 |
-
provider = "local"
|
| 197 |
-
else:
|
| 198 |
-
model_type = "gan"
|
| 199 |
-
provider = "local"
|
| 200 |
-
|
| 201 |
response = {
|
| 202 |
"status": "healthy",
|
| 203 |
"model_loaded": True,
|
| 204 |
-
"model_type":
|
| 205 |
-
"provider":
|
| 206 |
}
|
| 207 |
|
| 208 |
# Log API call
|
|
|
|
| 190 |
# -------------------------------------------------
|
| 191 |
@app.get("/health")
|
| 192 |
def health_check(request: Request):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 193 |
response = {
|
| 194 |
"status": "healthy",
|
| 195 |
"model_loaded": True,
|
| 196 |
+
"model_type": "hf_inference_api",
|
| 197 |
+
"provider": "fal-ai"
|
| 198 |
}
|
| 199 |
|
| 200 |
# Log API call
|