That guy James Bond :) commited on
Commit
d89e3d2
·
1 Parent(s): af61b34

Add root redirect to /docs for HF Spaces

Browse files
Files changed (1) hide show
  1. service/app.py +6 -1
service/app.py CHANGED
@@ -41,7 +41,7 @@ import numpy as np
41
  import onnxruntime as ort
42
  from fastapi import Depends, FastAPI, HTTPException, Query, Request, Security, status
43
  from fastapi.middleware.cors import CORSMiddleware
44
- from fastapi.responses import JSONResponse
45
  from fastapi.security import APIKeyHeader
46
  from pydantic import BaseModel, Field, field_validator
47
  from slowapi import Limiter, _rate_limit_exceeded_handler
@@ -298,6 +298,11 @@ app = FastAPI(
298
  openapi_url=None if DISABLE_DOCS else "/openapi.json",
299
  )
300
 
 
 
 
 
 
301
  # Add rate limiter to app
302
  app.state.limiter = limiter
303
  app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
 
41
  import onnxruntime as ort
42
  from fastapi import Depends, FastAPI, HTTPException, Query, Request, Security, status
43
  from fastapi.middleware.cors import CORSMiddleware
44
+ from fastapi.responses import JSONResponse, RedirectResponse
45
  from fastapi.security import APIKeyHeader
46
  from pydantic import BaseModel, Field, field_validator
47
  from slowapi import Limiter, _rate_limit_exceeded_handler
 
298
  openapi_url=None if DISABLE_DOCS else "/openapi.json",
299
  )
300
 
301
+ # Root redirect to Swagger docs
302
+ @app.get("/", include_in_schema=False)
303
+ async def root():
304
+ return RedirectResponse(url="/docs")
305
+
306
  # Add rate limiter to app
307
  app.state.limiter = limiter
308
  app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)