diff --git a/src/main.py b/src/main.py index 70adc13..29f0e84 100644 --- a/src/main.py +++ b/src/main.py @@ -33,7 +33,7 @@ def get_root(request: Request): if request.url.path != "/": raise HTTPException(status_code=404) - return renderTemplate("main.html") + return render("main.html") @app.post("/detect/{strength}") @@ -104,6 +104,5 @@ def get_models_for_detection_strength(strength: DetectionStrength): return ["db_resnet50", "vitstr_small"] -def renderTemplate(template, context={}): - template = env.get_template(template) - return HTMLResponse(template.render(context)) +def render(template, context={}): + return HTMLResponse(env.get_template(template).render(context))