add profile

1 files changed, 6 insertions, 3 deletions

diff --git a/app/models/image.py b/app/models/image.py
index e139123..b75b197 100644
--- a/app/models/image.py
+++ b/app/models/image.py
@@ -12,11 +12,12 @@ from sqlalchemy import (
     Table,
     func,
 )
-from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
+from sqlalchemy.orm import Mapped, mapped_column, relationship
 
-from .collections import Collection
+from app.utils.db import Base
+
+# from .collections import Collection
 
-Base = DeclarativeBase()
 
 image_tags = Table(
     "image_tags",
@@ -63,8 +64,10 @@ class Image(Base):
     )
 
 
+"""
 collections: Mapped[list["Collection"]] = relationship(
     "Collection",
     secondary="collection_images",
     back_populates="images",
 )
+"""