Added OmniGen Image to Image

app.py

@@ -228,7 +228,7 @@ def omnigen_image_to_image():
             # Save base64 image
             try:
                 image_path, image_url = save_base64_image(base64_image)
-                # image_url = "https://drive.google.com/uc?id=1JEHEy0zCVWOob4421hLQIPMbO_ebeCPS&export=download"
+                image_url = "https://drive.google.com/uc?id=1JEHEy0zCVWOob4421hLQIPMbO_ebeCPS&export=download"
             except Exception as e:
                 raise ValueError(f'Invalid base64 image data: {str(e)}')
 
@@ -243,7 +243,7 @@ def omnigen_image_to_image():
 
         # Modify workflow with inputs
         workflow["6"]["inputs"]["prompt"] = "in image_1 " + text_prompt
-        workflow["6"]["inputs"]["num_inference_steps"] = steps
+        workflow["6"]["inputs"]["num_inference_steps"] = 5
         workflow["12"]["inputs"]["url"] = image_url
 
         # WebSocket connection to queue the prompt