phuochungus commited on
Commit
eb371a0
1 Parent(s): 0c1187a

Add redirect to index route

Browse files
Files changed (1) hide show
  1. sources/Controllers/main.py +6 -1
sources/Controllers/main.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import yolov5
5
  from fastapi import File, UploadFile, status
6
  from fastapi.encoders import jsonable_encoder
7
- from fastapi.responses import JSONResponse
8
  from PIL import Image
9
  from vietocr.tool.config import Cfg
10
  from vietocr.tool.predictor import Predictor
@@ -49,6 +49,11 @@ config["predictor"]["beamsearch"] = False
49
  detector = Predictor(config)
50
 
51
 
 
 
 
 
 
52
  @app.post(
53
  "/uploader",
54
  )
 
4
  import yolov5
5
  from fastapi import File, UploadFile, status
6
  from fastapi.encoders import jsonable_encoder
7
+ from fastapi.responses import JSONResponse, RedirectResponse
8
  from PIL import Image
9
  from vietocr.tool.config import Cfg
10
  from vietocr.tool.predictor import Predictor
 
49
  detector = Predictor(config)
50
 
51
 
52
+ @app.get("/")
53
+ async def index():
54
+ return RedirectResponse(url="/docs")
55
+
56
+
57
  @app.post(
58
  "/uploader",
59
  )