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)}')
 
@@ -251,9 +251,6 @@ def omnigen_image_to_image():
         ws.connect(f"{ws_address}?clientId={client_id}&token={token}",
                    header={"Authorization": f"Bearer {token}"})
 
-        # Queue the prompt
-        prompt_id = queue_prompt(workflow, token)
-
         images = get_images(ws, workflow, token)
         ws.close()