mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-12-25 02:44:18 +00:00
Preliminary work for nlp model runtime integration
This commit is contained in:
parent
033872718e
commit
ad17ec2cc8
@ -25,8 +25,6 @@ def contributions():
|
||||
spacy_nlp_user_models = [
|
||||
x for x in current_user.spacy_nlp_pipeline_models
|
||||
]
|
||||
spacy_models = SpaCyNLPPipelineModel.query.all()
|
||||
print(spacy_models)
|
||||
return render_template(
|
||||
'contributions/contribution_overview.html.j2',
|
||||
tesseract_ocr_user_models=tesseract_ocr_user_models,
|
||||
|
@ -3,7 +3,8 @@ from app.models import (
|
||||
Job,
|
||||
JobResult,
|
||||
JobStatus,
|
||||
TesseractOCRPipelineModel
|
||||
TesseractOCRPipelineModel,
|
||||
SpaCyNLPPipelineModel
|
||||
)
|
||||
from datetime import datetime
|
||||
from flask import current_app
|
||||
@ -103,6 +104,16 @@ def _create_job_service(job):
|
||||
models_mount_target = f'/usr/local/share/tessdata/{model.filename}'
|
||||
models_mount = f'{models_mount_source}:{models_mount_target}:ro'
|
||||
mounts.append(models_mount)
|
||||
elif job.service == 'spacy-nlp-pipeline':
|
||||
model_id = hashids.decode(job.service_args['model'])
|
||||
model = SpaCyNLPPipelineModel.query.get(model_id)
|
||||
if model is None:
|
||||
job.status = JobStatus.FAILED
|
||||
return
|
||||
models_mount_source = model.path
|
||||
models_mount_target = f'/usr/local/share/spacy/models/{model.filename}'
|
||||
models_mount = f'{models_mount_source}:{models_mount_target}:ro'
|
||||
mounts.append(models_mount)
|
||||
''' ### Output mount ### '''
|
||||
output_mount_source = os.path.join(job.path, 'results')
|
||||
output_mount_target = '/output'
|
||||
|
Loading…
Reference in New Issue
Block a user