Update app.py

app.py

@@ -152,9 +152,6 @@ class AnimateController:
 
     def animate(
         self,
-        stable_diffusion_dropdown,
-        motion_module_dropdown,
-        base_model_dropdown,
         lora_alpha_slider,
         spatial_lora_slider,
         prompt_textbox,
@@ -167,12 +164,6 @@ class AnimateController:
         cfg_scale_slider,
         seed_textbox
     ):
-        if self.unet is None:
-            raise gr.Error(f"Please select a pretrained model path.")
-        if motion_module_dropdown == "":
-            raise gr.Error(f"Please select a motion module.")
-        if base_model_dropdown == "":
-            raise gr.Error(f"Please select a base DreamBooth model.")
 
         if is_xformers_available():
             self.unet.enable_xformers_memory_efficient_attention()
@@ -232,6 +223,10 @@ class AnimateController:
 
 controller = AnimateController()
 
+controller.update_stable_diffusion("stable-diffusion-v1-5")
+controller.update_motion_module("sd15_t2v_beta_motion.ckpt")
+controller.update_base_model("realistic2.safetensors")
+
 
 def ui():
     with gr.Blocks(css=css) as demo:
@@ -241,54 +236,21 @@ def ui():
             Fu-Yun Wang, Zhaoyang Huang (*Corresponding Author), Xiaoyu Shi, Weikang Bian, Guanglu Song, Yu Liu, Hongsheng Li (*Corresponding Author)<br>
             [arXiv Report](https://arxiv.org/abs/2402.00769) | [Project Page](https://animatelcm.github.io/) | [Github](https://github.com/G-U-N/AnimateLCM) | [Civitai](https://civitai.com/models/290375/animatelcm-fast-video-generation) | [Replicate](https://replicate.com/camenduru/animate-lcm)
             """
+            
+            '''
+            Important Notes: 
+            1. The generation speed is around 1~2 seconds. There is delay in the space.
+            2. Increase the sampling step and cfg if you want more fancy videos.
+            '''
         )
         with gr.Column(variant="panel"):
-            gr.Markdown(
-                """
-                ### 1. Model checkpoints (select pretrained model path first).
-                """
-            )
             with gr.Row():
-                stable_diffusion_dropdown = gr.Dropdown(
-                    label="Pretrained Model Path",
-                    choices=controller.stable_diffusion_list,
-                    # value="models/StableDiffusion/stable-diffusion-v1-5",
-                    interactive=True,
-                )
-                stable_diffusion_dropdown.change(fn=controller.update_stable_diffusion, inputs=[
-                                                 stable_diffusion_dropdown], outputs=[stable_diffusion_dropdown])
-
-                stable_diffusion_refresh_button = gr.Button(
-                    value="\U0001F503", elem_classes="toolbutton")
-
-                def update_stable_diffusion():
-                    controller.refresh_stable_diffusion()
-                    return gr.Dropdown.update(choices=controller.stable_diffusion_list)
-                stable_diffusion_refresh_button.click(
-                    fn=update_stable_diffusion, inputs=[], outputs=[stable_diffusion_dropdown])
-
-            with gr.Row():
-                motion_module_dropdown = gr.Dropdown(
-                    label="Select motion module",
-                    choices=controller.motion_module_list,
-                    interactive=True,
-                )
-                motion_module_dropdown.change(fn=controller.update_motion_module, inputs=[
-                                              motion_module_dropdown], outputs=[motion_module_dropdown])
-
-                motion_module_refresh_button = gr.Button(
-                    value="\U0001F503", elem_classes="toolbutton")
-
-                def update_motion_module():
-                    controller.refresh_motion_module()
-                    return gr.Dropdown.update(choices=controller.motion_module_list)
-                motion_module_refresh_button.click(
-                    fn=update_motion_module, inputs=[], outputs=[motion_module_dropdown])
 
                 base_model_dropdown = gr.Dropdown(
                     label="Select base Dreambooth model (required)",
                     choices=controller.personalized_model_list,
                     interactive=True,
+                    value="realistic2.safetensors"
                 )
                 base_model_dropdown.change(fn=controller.update_base_model, inputs=[
                                            base_model_dropdown], outputs=[base_model_dropdown])
@@ -328,9 +290,9 @@ def ui():
                 """
             )
 
-            prompt_textbox = gr.Textbox(label="Prompt", lines=2)
+            prompt_textbox = gr.Textbox(label="Prompt", lines=2, value="a boy holding a rabbit")
             negative_prompt_textbox = gr.Textbox(
-                label="Negative prompt", lines=2)
+                label="Negative prompt", lines=2, value="bad quality")
 
             with gr.Row().style(equal_height=False):
                 with gr.Column():
@@ -338,7 +300,7 @@ def ui():
                         sampler_dropdown = gr.Dropdown(label="Sampling method", choices=list(
                             scheduler_dict.keys()), value=list(scheduler_dict.keys())[0])
                         sample_step_slider = gr.Slider(
-                            label="Sampling steps", value=4, minimum=1, maximum=25, step=1)
+                            label="Sampling steps", value=6, minimum=1, maximum=25, step=1)
 
                     width_slider = gr.Slider(
                         label="Width",            value=512, minimum=256, maximum=1024, step=64)
@@ -347,7 +309,7 @@ def ui():
                     length_slider = gr.Slider(
                         label="Animation length", value=16,  minimum=12,   maximum=20,   step=1)
                     cfg_scale_slider = gr.Slider(
-                        label="CFG Scale",        value=1, minimum=1,   maximum=2)
+                        label="CFG Scale",        value=1.5, minimum=1,   maximum=2)
 
                     with gr.Row():
                         seed_textbox = gr.Textbox(label="Seed", value=-1)
@@ -365,9 +327,6 @@ def ui():
             generate_button.click(
                 fn=controller.animate,
                 inputs=[
-                    stable_diffusion_dropdown,
-                    motion_module_dropdown,
-                    base_model_dropdown,
                     lora_alpha_slider,
                     spatial_lora_slider,
                     prompt_textbox,
@@ -386,6 +345,7 @@ def ui():
     return demo
 
 
+
 if __name__ == "__main__":
     demo = ui()
     # gr.close_all()