Jon Solow commited on
Commit
e35ad6d
·
1 Parent(s): 79cf8f5

Set main url to docs

Browse files
Files changed (1) hide show
  1. src/main.py +4 -3
src/main.py CHANGED
@@ -1,14 +1,15 @@
1
  from fastapi import FastAPI, UploadFile, File, HTTPException, status, Header
 
2
 
3
  from handler import handle_file, handle_url
4
  from predict import predict_model
5
 
6
- app = FastAPI()
7
 
 
8
 
9
  @app.get("/")
10
- async def root():
11
- return {"message": "Hello World"}
12
 
13
 
14
  @app.get("/predict_url")
 
1
  from fastapi import FastAPI, UploadFile, File, HTTPException, status, Header
2
+ from fastapi.responses import RedirectResponse
3
 
4
  from handler import handle_file, handle_url
5
  from predict import predict_model
6
 
 
7
 
8
+ app = FastAPI()
9
 
10
  @app.get("/")
11
+ async def docs_redirect():
12
+ return RedirectResponse(url='/docs')
13
 
14
 
15
  @app.get("/predict_url")