Update app.py

app.py

@@ -288,9 +288,14 @@ def get_duration(input_image, prompt, n_prompt, seed, total_second_length, laten
     return total_second_length * 60
 
 @spaces.GPU(duration=get_duration)
-def process(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf):
+def process(input_image, prompt, t2v, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf):
     global stream
-    assert input_image is not None, 'No input image!'
+    
+    # assert input_image is not None, 'No input image!'
+    if t2v:
+        default_height, default_width = 640, 640
+        input_image = np.ones((default_height, default_width, 3), dtype=np.uint8) * 255
+        print("No input image provided. Using a blank white image.") 
 
     yield None, None, '', '', gr.update(interactive=False), gr.update(interactive=True)
 
@@ -331,10 +336,13 @@ css = make_progress_bar_css()
 block = gr.Blocks(css=css).queue()
 with block:
     gr.Markdown('# FramePack-F1')
+    gr.Markdown('Demo for FramePack F1 - a FramePack model that only predicts future frames from history frames')
+    gr.Markdown('adapted from the officical code repo [FramePack](https://github.com/lllyasviel/FramePack) by lllyasviel and (FramePack Studio)[https://github.com/colinurbs/FramePack-Studio]')
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(sources='upload', type="numpy", label="Image", height=320)
             prompt = gr.Textbox(label="Prompt", value='')
+            t2v = gr.Checkbox(label="do text-to-video", value=False)
             example_quick_prompts = gr.Dataset(samples=quick_prompts, label='Quick List', samples_per_page=1000, components=[prompt])
             example_quick_prompts.click(lambda x: x[0], inputs=[example_quick_prompts], outputs=prompt, show_progress=False, queue=False)
 
@@ -370,7 +378,7 @@ with block:
 
     gr.HTML('<div style="text-align:center; margin-top:20px;">Share your results and find ideas at the <a href="https://x.com/search?q=framepack&f=live" target="_blank">FramePack Twitter (X) thread</a></div>')
 
-    ips = [input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf]
+    ips = [input_image, prompt, t2v, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf]
     start_button.click(fn=process, inputs=ips, outputs=[result_video, preview_image, progress_desc, progress_bar, start_button, end_button])
     end_button.click(fn=end_process)