cerbervs/doctr-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: model name

Browse Source
This commit is contained in:
Mark Bailey 2024-10-28 11:23:51 -04:00
parent 129802f290
commit f8c26830c9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.py
View File

@ -101,7 +101,7 @@ def get_models_for_detection_strength(strength: DetectionStrength):
elif strength == DetectionStrength.medium: elif strength == DetectionStrength.medium:
return ["db_resnet50", "vitstr_base"] return ["db_resnet50", "vitstr_base"]
elif strength == DetectionStrength.low: elif strength == DetectionStrength.low:
return ["db_resnet50", "vitstr_tiny"] return ["db_resnet50", "vitstr_small"]
def renderTemplate(template, context={}): def renderTemplate(template, context={}):