Update app.py

app.py

@@ -173,8 +173,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 output_image = gr.Image(label="Output", interactive=False, elem_id="output", type="filepath")
                 use_image_button = gr.Button("♻️ Use this Image for Next Edit")
                 create_video_button = gr.Button("Create a video between the two 🎥", variant="primary", visible=False)
-                video_output = gr.Video(label="Generated Video", visible=False, show_download_button=True)
-                
+                with gr.Group(visible=False) as video_group:
+                    video_output = gr.Video(label="Generated Video", show_download_button=True)
+                    gr.Markdown("Video Generated by [Wan 2.2 first-last-frame](https://huggingface.co/spaces/multimodalart/wan-2-2-first-last-frame)")
         gr.Markdown("## Thank you for being a PRO! 🤗")
     
     login_button = gr.LoginButton()
@@ -184,11 +185,11 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
         triggers=[generate_button.click, prompt_input.submit],
         fn=lambda: [gr.update(visible=False), gr.update(visible=False)],
         inputs=[],
-        outputs=[create_video_button, video_output],
+        outputs=[create_video_button, video_group],
     ).then(
         fn=unified_image_generator,
         inputs=[prompt_input, image_input_gallery],
-        outputs=[output_image, create_video_button, video_output],
+        outputs=[output_image, create_video_button, video_group],
     )
 
     use_image_button.click(
@@ -201,7 +202,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     create_video_button.click(
         fn=lambda: gr.update(visible=True),
         inputs=[],
-        outputs=[video_output],
+        outputs=[video_group],
     ).then(
         fn=create_video_transition,
         inputs=[image_input_gallery, output_image],