sivakorn-su
commited on
Commit
·
18a8fe5
1
Parent(s):
2c9f05b
fix code
Browse files
app.py
CHANGED
@@ -92,9 +92,10 @@ app.add_middleware(
|
|
92 |
allow_headers=["*"],
|
93 |
)
|
94 |
|
95 |
-
|
96 |
-
|
97 |
-
|
|
|
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 {
|