From f8c26830c98f73b8c43fb2828bcabb2be8907b81 Mon Sep 17 00:00:00 2001 From: Mark Bailey Date: Mon, 28 Oct 2024 11:23:51 -0400 Subject: [PATCH] fix: model name --- src/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/main.py b/src/main.py index c7091d5..70adc13 100644 --- a/src/main.py +++ b/src/main.py @@ -101,7 +101,7 @@ def get_models_for_detection_strength(strength: DetectionStrength): elif strength == DetectionStrength.medium: return ["db_resnet50", "vitstr_base"] elif strength == DetectionStrength.low: - return ["db_resnet50", "vitstr_tiny"] + return ["db_resnet50", "vitstr_small"] def renderTemplate(template, context={}):