from fastapi import FastAPI from fastapi.staticfiles import StaticFiles from models import CheckResponse from completions import check_text, load_model app = FastAPI() model, tokenizer, device = load_model() @app.get("/check", response_model=CheckResponse) def check(text: str): return CheckResponse(text=text, words=check_text(text, model, tokenizer, device)) app.mount("/", StaticFiles(directory="frontend/public", html=True))