Added OmniGen Image to Image

app.py

@@ -193,6 +193,7 @@ def omnigen_image_to_image():
         return jsonify({'error': 'Text prompt is required'}), 400
 
     steps = data.get('steps')
+
     if not steps:
         steps = 50
 
@@ -228,7 +229,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 +244,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"] = 5
+        workflow["6"]["inputs"]["num_inference_steps"] = steps
         workflow["12"]["inputs"]["url"] = image_url
 
         # WebSocket connection to queue the prompt