birbnetes/model-service
4
0
Fork 0
Code Issues Releases Wiki Activity

Added ability to store CNN models
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell
2020-07-28 20:19:52 +02:00
parent fe965706b5
commit 284dfce735
12 changed files with 224 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
model_service/model/default.py
View File

@ -2,7 +2,11 @@
from .db import db
from sqlalchemy.dialects.postgresql import UUID
from .aimodel import AIModelType
class Default(db.Model):
aimodel_id = db.Column(UUID(as_uuid=True), db.ForeignKey("ai_model.id"), nullable=False, primary_key=True)
type = db.Column(db.Enum(AIModelType), unique=True, nullable=False, primary_key=True)
aimodel_id = db.Column(UUID(as_uuid=True), db.ForeignKey("ai_model.id"), nullable=False)
aimodel = db.relationship("AIModel", backref=db.backref("default", lazy=True, cascade="save-update, merge, delete, delete-orphan"))