From d48f5663896639c22f1b200e5281bac7d8167a8f Mon Sep 17 00:00:00 2001
From: ManInDark <61268856+ManInDark@users.noreply.github.com>
Date: Fri, 4 Aug 2023 23:22:24 +0200
Subject: [PATCH] Moved some stuff

---
 ui/easydiffusion/utils/save_utils.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/ui/easydiffusion/utils/save_utils.py b/ui/easydiffusion/utils/save_utils.py
index 0082d7f2..63d00494 100644
--- a/ui/easydiffusion/utils/save_utils.py
+++ b/ui/easydiffusion/utils/save_utils.py
@@ -159,8 +159,7 @@ def save_images_to_disk(
             from easydiffusion.easydb.database import SessionLocal
 
             session = SessionLocal()
-            
-            img = Image(
+            session.add(Image(
                 path = path_i,
                 seed = metadata_entries[i]["seed"],
                 use_stable_diffusion_model = metadata_entries[i]["use_stable_diffusion_model"],
@@ -176,9 +175,7 @@ def save_images_to_disk(
                 tiling = metadata_entries[i]["tiling"],
                 use_face_correction = metadata_entries[i]["use_face_correction"],
                 use_upscale = metadata_entries[i]["use_upscale"]
-            )
-
-            session.add(img)
+            ))
             session.commit()
             session.close()