Add SpaCyNLPPipelineModel to flask shell context

This commit is contained in:
Patrick Jentsch 2022-10-21 13:33:25 +02:00
parent 8bee4196d3
commit cbf3abb424

View File

@ -13,6 +13,7 @@ from app.models import (
Permission, Permission,
Role, Role,
TesseractOCRPipelineModel, TesseractOCRPipelineModel,
SpaCyNLPPipelineModel,
User User
) # noqa ) # noqa
from flask import Flask # noqa from flask import Flask # noqa
@ -42,6 +43,7 @@ def make_shell_context() -> Dict[str, Any]:
'Permission': Permission, 'Permission': Permission,
'Role': Role, 'Role': Role,
'TesseractOCRPipelineModel': TesseractOCRPipelineModel, 'TesseractOCRPipelineModel': TesseractOCRPipelineModel,
'SpaCyNLPPipelineModel': SpaCyNLPPipelineModel,
'User': User 'User': User
} }