modified: app.py

app.py

@@ -4,8 +4,9 @@ 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)
-    client = Client("black-forest-labs/FLUX.1-Kontext-Dev", headers={"x-ip-token": x_ip_token})
-    return client
+    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
 
 def text_to_image(prompt, image, client):
     result = client.predict(
@@ -21,11 +22,12 @@ def text_to_image(prompt, image, client):
     return result
 
 with gr.Blocks() as demo:
-    client = gr.State(None)
+    flux_client = gr.State(None)
+    ltx_client = gr.State(None)
     image = gr.Image()
     input_image = gr.Image(type="filepath")
     prompt = gr.Textbox(max_lines=1)
-    prompt.submit(text_to_image, [prompt, input_image, client], [image])
+    prompt.submit(text_to_image, [prompt, input_image, flux_client], [image])
 
     # gr.Examples(
     #     examples=[
@@ -35,7 +37,7 @@ with gr.Blocks() as demo:
     #     outputs=image,
     # )
 
-    demo.load(set_client_for_session, None, client)
+    demo.load(set_client_for_session, None, [flux_client, ltx_client])
 
 demo.launch()