LogicGoInfotechSpaces commited on
Commit
58427fc
·
1 Parent(s): 9587da3

Update health endpoint to match requested format with hf_inference_api and fal-ai

Browse files
Files changed (1) hide show
  1. 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": model_type,
205
- "provider": 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