modify req

app.py

@@ -99,6 +99,10 @@ def update_preview_video(checkpoint_dir):
     return gr.update(value=f'results/{parent_dir}/source.mp4')
 
 
+def update_generated_prompt(text):
+    return gr.update(value=text)
+
+
 if __name__ == "__main__":
 
     if os.path.exists('results/custom'):
@@ -130,8 +134,67 @@ if __name__ == "__main__":
         ['results/train_ride/source.mp4','A motorbike driving in a forest','object','train_ride/checkpoint'], 
     ]
 
+    gradio_theme = gr.themes.Default()
     # 创建Gradio界面
-    with gr.Blocks() as demo:
+    with gr.Blocks(
+        theme=gradio_theme,
+        title="Motion Inversion",
+        css="""
+            #download {
+                height: 118px;
+            }
+            .slider .inner {
+                width: 5px;
+                background: #FFF;
+            }
+            .viewport {
+                aspect-ratio: 4/3;
+            }
+            .tabs button.selected {
+                font-size: 20px !important;
+                color: crimson !important;
+            }
+            h1 {
+                text-align: center;
+                display: block;
+            }
+            h2 {
+                text-align: center;
+                display: block;
+            }
+            h3 {
+                text-align: center;
+                display: block;
+            }
+            .md_feedback li {
+                margin-bottom: 0px !important;
+            }
+        """,
+        head="""
+            <script async src="https://www.googletagmanager.com/gtag/js?id=G-1FWSVCGZTG"></script>
+            <script>
+                window.dataLayer = window.dataLayer || [];
+                function gtag() {dataLayer.push(arguments);}
+                gtag('js', new Date());
+                gtag('config', 'G-1FWSVCGZTG');
+            </script>
+        """,
+    ) as demo:
+        gr.Markdown(
+            """
+            # Motion Inversion for Video Customization
+            <p align="center">
+            <a title="Page" href="https://wileewang.github.io/MotionInversion/" target="_blank" rel="noopener noreferrer" style="display: inline-block;">
+                <img src="https://img.shields.io/badge/Project-Website-pink?logo=googlechrome&logoColor=white">
+            </a>
+            <a title="arXiv" href="https://arxiv.org/abs/2403.20193" target="_blank" rel="noopener noreferrer" style="display: inline-block;">
+                <img src="https://img.shields.io/badge/arXiv-Paper-b31b1b?logo=arxiv&logoColor=white">
+            </a>
+            <a title="Github" href="https://github.com/EnVision-Research/MotionInversion" target="_blank" rel="noopener noreferrer" style="display: inline-block;">
+                <img src="https://img.shields.io/github/stars/EnVision-Research/MotionInversion?label=GitHub%20%E2%98%85&logo=github&color=C8C" alt="badge-github-stars">
+            </a>
+        """
+        )
         with gr.Tab("Train"):
             with gr.Row():
                 with gr.Column():
@@ -169,6 +232,7 @@ if __name__ == "__main__":
                 with gr.Column():
                     
                     output_video = gr.Video(label="Output Video")
+                    generated_prompt = gr.Textbox(label="Generated Prompt")
 
             with gr.Accordion("Advanced Settings", open=False):
                 with gr.Row():
@@ -184,6 +248,6 @@ if __name__ == "__main__":
             checkpoint_dropdown.change(fn=update_preview_video, inputs=checkpoint_dropdown, outputs=preview_video)
             checkpoint_output.change(update_checkpoints, inputs=checkpoint_output, outputs=checkpoint_dropdown)
             inference_button.click(inference_model, inputs=[text_input, checkpoint_dropdown,inference_steps,motion_type, seed], outputs=output_video)
-
+            output_video.change(fn=update_generated_prompt, inputs=[text_input], outputs=generated_prompt)
     # 启动Gradio界面
     demo.launch()

requirements.txt

@@ -43,8 +43,8 @@ requests==2.32.3
 safetensors==0.4.5
 sympy==1.13.3
 tokenizers==0.20.1
-torch==2.4.1
-torchvision==0.19.1
+# torch==2.4.1
+# torchvision==0.19.1
 tqdm==4.66.5
 transformers==4.45.2
 triton==3.0.0