Changes to be committed:

app.py

@@ -1,13 +1,6 @@
 import gradio as gr
 from gradio_client import Client, handle_file
 
-def set_client_for_session(request: gr.Request):
-    x_ip_token = request.headers['x-ip-token']
-    print(x_ip_token)
-    flux_client = Client("black-forest-labs/FLUX.1-Kontext-Dev", headers={"x-ip-token": x_ip_token})
-    ltx_client = Client("KingNish/ltx-video-distilled", headers={"x-ip-token": x_ip_token})
-    return flux_client, ltx_client
-
 def image_to_image(prompt, image, flux_client):
     prompt = "You are an AI image-to-image editor that transforms a user-provided input image based on their prompt while maintaining consistency and fidelity to the original image. Generate an output image that accurately reflects the user's requested modifications, preserving key elements like style, composition, and context from the input image unless explicitly instructed otherwise. " + prompt
     image = flux_client.predict(
@@ -34,22 +27,24 @@ def image_to_video(prompt, image, ltx_client):
     print(result)
     return result
 
-def personalized_video(prompt, image, flux_client, ltx_client):
+def personalized_video(prompt, image, flux_client, ltx_client, request: gr.Request):
+    x_ip_token = request.headers['x-ip-token']
+    flux_client = Client("black-forest-labs/FLUX.1-Kontext-Dev", headers={"x-ip-token": x_ip_token})
+    ltx_client = Client("KingNish/ltx-video-distilled", headers={"x-ip-token": x_ip_token})
+    print(x_ip_token)
     image = image_to_image(prompt, image, flux_client)
     yield image, None
     video = image_to_video(prompt, image, ltx_client)
     yield image, video
 
 with gr.Blocks() as demo:
-    flux_client = gr.State(None)
-    ltx_client = gr.State(None)
     input_image = gr.Image(type="filepath")
     prompt = gr.Textbox(max_lines=1)
     edited_image = gr.Image()
     output_video = gr.Video()
     submit_button = gr.Button("Submit")
-    submit_button.click(personalized_video, [prompt, input_image, flux_client, ltx_client], [edited_image, output_video])
-    prompt.submit(personalized_video, [prompt, input_image, flux_client, ltx_client], [edited_image, output_video])
+    submit_button.click(personalized_video, [prompt, input_image], [edited_image, output_video])
+    prompt.submit(personalized_video, [prompt, input_image], [edited_image, output_video])
 
     # gr.Examples(
     #     examples=[
@@ -62,7 +57,5 @@ with gr.Blocks() as demo:
     #     cache_mode = "eager"
     # )
 
-    demo.load(set_client_for_session, None, [flux_client, ltx_client])
-
 demo.launch()