rahul7star commited on
Commit
e87ad32
·
verified ·
1 Parent(s): 810b94d

Update autorun_lora_gradio.py

Browse files
Files changed (1) hide show
  1. autorun_lora_gradio.py +8 -0
autorun_lora_gradio.py CHANGED
@@ -47,6 +47,14 @@ app.add_middleware(
47
  def health_check():
48
  return {"status": "✅ FastAPI running on Hugging Face Spaces!"}
49
 
 
 
 
 
 
 
 
 
50
 
51
 
52
  REPO_ID = "rahul7star/ohamlab"
 
47
  def health_check():
48
  return {"status": "✅ FastAPI running on Hugging Face Spaces!"}
49
 
50
+ @app.get("/healthz")
51
+ def healthz():
52
+ return {"ok": True}
53
+
54
+ @app.get("/docs", include_in_schema=False)
55
+ def custom_docs():
56
+ return JSONResponse(get_openapi(title="LoRA Autorun API", version="1.0.0", routes=app.routes))
57
+
58
 
59
 
60
  REPO_ID = "rahul7star/ohamlab"