Added OmniGen Image to Image

app.py

@@ -273,7 +273,7 @@ def omnigen_image_to_image():
                 img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
                 output_images_base64.append(img_str)
 
-        return jsonify({'images': output_images_base64, 'message': 'Image generated successfully'}), 200
+        return jsonify({'images': output_images_base64}), 200
 
     except Exception as e:
         return jsonify({'message': 'Unable to connect to the server. Make sure the server is running', 'error': str(e)}), 500