diff --git a/kotti_ai/resources.py b/kotti_ai/resources.py index bba4b2d..e17b5d7 100644 --- a/kotti_ai/resources.py +++ b/kotti_ai/resources.py @@ -95,6 +95,8 @@ class AImage(SaveDataMixin, Content): #: Primary key column in the DB #: (:class:`sqlalchemy.types.Integer`) id = Column(Integer(), ForeignKey("contents.id"), primary_key=True) + from sqlalchemy import String + out = Column(String(500)) # add for ai predict out data_filters = ( WithThumbnailFilter(size=(128, 128), format="PNG"), @@ -112,4 +114,8 @@ class AImage(SaveDataMixin, Content): ], ) + def __init__(self, out=0, **kwargs): + super(AImage, self).__init__(**kwargs) + self.out = out +