Fix TranskribusHTRModel display problems

This commit is contained in:
Patrick Jentsch
2022-10-04 15:33:51 +02:00
parent 1c8f4f7e9c
commit ca27f2e6f2
4 changed files with 48 additions and 5 deletions

View File

@ -10,7 +10,11 @@ from wtforms import (
ValidationError
)
from wtforms.validators import InputRequired, Length
from app.models import TesseractOCRModel, TranskribusHTRModel
from app.models import (
TRANSKRIBUS_HTR_MODELS,
TesseractOCRModel,
TranskribusHTRModel
)
from . import SERVICES
@ -119,7 +123,7 @@ class CreateTranskribusHTRPipelineJobForm(CreateJobBaseForm):
if x.shared == True or x.user == current_user
]
self.model.choices = [('', 'Choose your option')]
self.model.choices += [(x.hashid, x.transkribus_name) for x in models]
self.model.choices += [(x.hashid, [y['name'] for y in TRANSKRIBUS_HTR_MODELS if y['modelId'] == x.transkribus_model_id ][0]) for x in models]
self.model.default = ''
self.version.choices = [(x, x) for x in service_manifest['versions']]
self.version.data = version