Changes to be committed:

modified: app.py

Changes not staged for commit:
modified: app.py

app.py

@@ -1,36 +1,39 @@
 import gradio as gr
 from gradio_client import Client, handle_file
 
-def text_to_image(prompt, image, request: gr.Request):
+def set_client_for_session(request: gr.Request):
     x_ip_token = request.headers['x-ip-token']
-    print(x_ip_token)
     client = Client("black-forest-labs/FLUX.1-Kontext-Dev", headers={"x-ip-token": x_ip_token})
+    return client
+
+def text_to_image(prompt, image, client):
     result = client.predict(
-            input_image=handle_file(image),
-            prompt=prompt,
-            seed=0,
-            randomize_seed=True,
-            guidance_scale=2.5,
-            steps=30,
-            api_name="/infer"
+        input_image=handle_file(image),
+        prompt=prompt,
+        seed=0,
+        randomize_seed=True,
+        guidance_scale=2.5,
+        steps=30,
+        api_name="/infer"
     )
     print(result)
     return result
 
-
 with gr.Blocks() as demo:
     image = gr.Image()
     input_image = gr.Image(type="filepath")
     prompt = gr.Textbox(max_lines=1)
     prompt.submit(text_to_image, [prompt, input_image], [image])
 
-    gr.Examples(
-        examples=[
-            ["A sunny day", "https://black-forest-labs-flux-1-kontext-dev.hf.space/gradio_api/file=/tmp/gradio/927b05b08b8ec2201f7d95722471d2089d4197f152c9c07d0bc9f9672962d03b/flowers.png"],
-        ],
-        inputs=[prompt, input_image],
-        outputs=image,
-    )
+    # gr.Examples(
+    #     examples=[
+    #         ["A sunny day", "https://black-forest-labs-flux-1-kontext-dev.hf.space/gradio_api/file=/tmp/gradio/927b05b08b8ec2201f7d95722471d2089d4197f152c9c07d0bc9f9672962d03b/flowers.png"],
+    #     ],
+    #     inputs=[prompt, input_image],
+    #     outputs=image,
+    # )
+
+    demo.load(set_client_for_session, None, client)
 
 demo.launch()