RosiYo commited on
Commit
b59078e
·
1 Parent(s): b08b610

feat[bert]: FastAPI interface

Browse files
Files changed (1) hide show
  1. ml-service/app.py +1 -1
ml-service/app.py CHANGED
@@ -7,7 +7,7 @@ import logging
7
  logging.basicConfig(level = logging.INFO)
8
 
9
  # Create the FastAPI instance
10
- app = FastAPI()
11
  classifier = Classifier()
12
 
13
  @app.post("/predict")
 
7
  logging.basicConfig(level = logging.INFO)
8
 
9
  # Create the FastAPI instance
10
+ app = FastAPI(docs_url='/', redoc_url='/new_redoc')
11
  classifier = Classifier()
12
 
13
  @app.post("/predict")