Pushing changes to database session

users/routes.py

@@ -2,6 +2,7 @@ from fastapi import APIRouter, status, Depends, Request, UploadFile, File, HTTPE
 from fastapi.responses import JSONResponse
 from sqlalchemy.orm import Session
 from core.database import get_db
+from users.models import UserEmbeddings
 from core.security import get_current_user, create_refresh_token, create_access_token
 from users.schemas import UserCreate, UserBase, UserEmbeddingsBase, User, UserUpdate
 from users.services import create_user_account, create_user_embeddings, update_user, update_user_embeddings, get_user_by_id, get_user_by_email
@@ -60,7 +61,7 @@ async def create_face_embeddings(file: UploadFile = File(...), user: User = Depe
 
 @router.get("/me/face/", status_code=status.HTTP_200_OK)
 async def get_face_embeddings(user: User = Depends(get_current_user), db: Session = Depends(get_db)):
-    face = db.query(UserEmbeddingsBase).filter(UserEmbeddingsBase.user_id == user.id).first()
+    face = db.query(UserEmbeddings).filter(UserEmbeddings.user_id == user.id).first()
     if not face:
         raise HTTPException(status_code=404, detail="Face embeddings not found")
     return JSONResponse(content={"embeddings": face.embeddings}, status_code=status.HTTP_200_OK)