add swagger under nginx

lm-cyber · Apr 9, 2024 · 1c74159 · 1c74159
1 parent f2ee198
commit 1c74159
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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