Spaces:
Sleeping
Sleeping
Update endpoints.py
Browse files- 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):
|