sivakorn-su commited on
Commit
18a8fe5
·
1 Parent(s): 2c9f05b
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -92,9 +92,10 @@ app.add_middleware(
92
  allow_headers=["*"],
93
  )
94
 
95
- @app.get("/health")
96
- async def health_check():
97
- return {
 
98
  "status": "running",
99
  "models_loaded": {
100
  "pipelines": len(pipelines),
@@ -104,10 +105,6 @@ async def health_check():
104
  "cuda_devices": torch.cuda.device_count() if torch.cuda.is_available() else 0
105
  }
106
 
107
- @app.get("/")
108
- def check_api():
109
- return {"message": "API is up and running"}
110
-
111
  @app.get("/key")
112
  async def check_env():
113
  return {
 
92
  allow_headers=["*"],
93
  )
94
 
95
+
96
+ @app.get("/")
97
+ def check_api():
98
+ return {
99
  "status": "running",
100
  "models_loaded": {
101
  "pipelines": len(pipelines),
 
105
  "cuda_devices": torch.cuda.device_count() if torch.cuda.is_available() else 0
106
  }
107
 
 
 
 
 
108
  @app.get("/key")
109
  async def check_env():
110
  return {