serenarolloh commited on
Commit
8cf136d
·
verified ·
1 Parent(s): 447c004

Update endpoints.py

Browse files
Files changed (1) hide show
  1. endpoints.py +6 -0
endpoints.py CHANGED
@@ -7,8 +7,14 @@ import json
7
  import os
8
  from config import settings
9
  import utils
 
10
  from routers.donut_inference import process_document_donut
11
 
 
 
 
 
 
12
  router = APIRouter()
13
 
14
  def count_values(obj):
 
7
  import os
8
  from config import settings
9
  import utils
10
+ from routers import inference, training
11
  from routers.donut_inference import process_document_donut
12
 
13
+ app = FastAPI(openapi_url="/api/v1/sparrow-ml/openapi.json", docs_url="/api/v1/sparrow-ml/docs")
14
+
15
+ app.include_router(inference.router, prefix="/api-inference/v1/sparrow-ml", tags=["Inference"])
16
+ app.include_router(training.router, prefix="/api-training/v1/sparrow-ml", tags=["Training"])
17
+
18
  router = APIRouter()
19
 
20
  def count_values(obj):