File size: 486 Bytes
ffcf8df | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from fastapi import APIRouter, HTTPException, Request
from src.schemas.requests import FixNewlinesRequest
from src.schemas.responses import FixNewlinesResponse
router = APIRouter()
@router.post("/fix-newlines", response_model=FixNewlinesResponse)
def fix_newlines(request: Request, body: FixNewlinesRequest):
pipeline = request.app.state.one_model_pipeline
fixed = pipeline.predict(body.text)
return FixNewlinesResponse(fixed_text=fixed, model_used=pipeline.model_name)
|