KingNish's picture
Update app.py
a7feda8 verified
raw
history blame contribute delete
943 Bytes
import gradio as gr
from gradio_client import Client, handle_file
def text_to_image(client, prompt, image, duration):
result = client.predict(
prompt=prompt,
input_image_filepath=handle_file(image),
duration_ui=duration,
api_name="/generate_video"
)
return result
def set_client_for_session(request: gr.Request):
x_ip_token = request.headers['x-ip-token']
# The "gradio/text-to-image" space is a ZeroGPU space
return Client("KingNish/ltx-video-distilled", headers={"X-IP-Token": x_ip_token})
with gr.Blocks() as demo:
client = gr.State()
image = gr.Image(type="filepath")
prompt = gr.Textbox(max_lines=1)
duration = gr.Slider(minimum=1, maximum=8, step=1, interactive=True)
submit_button = gr.Button("Submit")
output = gr.Video()
submit_button.click(text_to_image, [client, prompt, image, duration], [output])
demo.load(set_client_for_session, None, client)
demo.launch()