Update ominicontrol.py

ominicontrol.py

@@ -47,7 +47,9 @@ def generate_image(
     steps,
     use_random_seed,
     seed,
-):
+):  
+    condition_id = log_image(image)
+
     # Prepare Condition
     def resize(img, factor=16):
         w, h = img.size
@@ -120,8 +122,6 @@ def generate_image(
         max_sequence_length=32,
     ).images[0]
     # result_img = image
-
-    condition_id = log_image(result_img)
     result_id = log_image(result_img)
 
     log_data = {