Update ominicontrol.py

ominicontrol.py

@@ -6,7 +6,7 @@ import random
 
 from OminiControl.src.flux.generate import generate, seed_everything
 
-from log import insert_log, log_image
+#from log import insert_log, log_image
 
 print("Loading model...")
 pipe = FluxPipeline.from_pretrained(
@@ -126,6 +126,7 @@ def generate_image(
     # result_img = image
     result_id = log_image(result_img)
 
+    '''
     log_data = {
         "condition": condition_id,
         "result": result_id,
@@ -141,6 +142,8 @@ def generate_image(
     log_data = {k: str(v) for k, v in log_data.items()}
 
     _, log_id = insert_log("inference", log_data)
+    '''
+    log_id = 0
 
     print(f"Image log ID: {log_id}")
 
@@ -151,6 +154,7 @@ def vote_feedback(
     log_id,
     feedback,
 ):
+    '''
     log_data = {
         "log_id": log_id,
         "feedback": feedback,
@@ -158,3 +162,5 @@ def vote_feedback(
     log_data = {k: str(v) for k, v in log_data.items()}
 
     insert_log("feedback", log_data)
+    '''
+    pass