From 0af50c21ae366c82eaf90c6d5d8195fd6f257003 Mon Sep 17 00:00:00 2001 From: Mark Bailey Date: Mon, 4 Nov 2024 15:29:05 -0500 Subject: [PATCH] refactor: render --- src/main.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) 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))