File size: 664 Bytes
ffcf8df | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | from fastapi import APIRouter, HTTPException, Request
from src.schemas.requests import FixNewlinesAllModelsRequest
from src.schemas.responses import FixNewlinesAllModelsResponse, ModelResult
router = APIRouter()
@router.post("/fix-newlines-all-models", response_model=FixNewlinesAllModelsResponse)
def fix_newlines_all_models(request: Request, body: FixNewlinesAllModelsRequest):
pipeline = request.app.state.all_models_pipeline
results_dict = pipeline.predict(body.text)
results = [
ModelResult(model_name=name, fixed_text=text)
for name, text in results_dict.items()
]
return FixNewlinesAllModelsResponse(results=results)
|