#!/usr/bin/env python3
from marshmallow import fields
from marshmallow_sqlalchemy import ModelSchema
from model import AIModel


class AIModelSchema(ModelSchema):
    class Meta:
        model = AIModel