Integrate TranskribusHTRModels

This commit is contained in:
Patrick Jentsch
2022-04-22 15:27:52 +02:00
parent 9d4001f469
commit ccdd0d3faa
13 changed files with 409 additions and 25 deletions

@ -1,5 +1,11 @@
from app import db
from app.models import Job, JobResult, JobStatus, TesseractOCRModel
from app.models import (
Job,
JobResult,
JobStatus,
TesseractOCRModel,
TranskribusHTRModel
)
from datetime import datetime
from flask import current_app
from werkzeug.utils import secure_filename
@ -56,7 +62,8 @@ class CheckJobsMixin:
if 'binarization' in job.service_args and job.service_args['binarization']:
command += ' --binarize'
elif job.service == 'transkribus-htr-pipeline':
command += f' -m {job.service_args["model"]}'
transkribus_htr_model = TranskribusHTRModel.query.get(job.service_args['model'])
command += f' -m {transkribus_htr_model.transkribus_model_id}'
readcoop_username = current_app.config.get('NOPAQUE_READCOOP_USERNAME')
command += f' --readcoop-username "{readcoop_username}"'
readcoop_password = current_app.config.get('NOPAQUE_READCOOP_PASSWORD')