diff --git a/app/main.py b/app/main.py index fb4cd26..03200a5 100644 --- a/app/main.py +++ b/app/main.py @@ -1,14 +1,12 @@ import os import sys - import uvicorn from fastapi import FastAPI, Request from fastapi.logger import logger from fastapi.exceptions import RequestValidationError from fastapi.middleware.cors import CORSMiddleware - from model import Model from predict import predict from config import CONFIG @@ -22,6 +20,10 @@ terms_of_service=None, contact=None, license_info=None, + docs_url='/api/docs', + redoc_url='/api/redoc', + openapi_url='/api/openapi.json' + ) # TODO change