更新 'kotti_ai/resources.py'

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
+