Set default pretrained model to "checkpoints/LayerAnimate-Mix"

app.py

@@ -68,6 +68,7 @@ class LayerAnimate:
         self.sample_grid = np.meshgrid(np.linspace(0, self.W - 1, 10, dtype=int), np.linspace(0, self.H - 1, 10, dtype=int))
         self.sample_grid = np.stack(self.sample_grid, axis=-1).reshape(-1, 1, 2)
         self.sample_grid = np.repeat(self.sample_grid, self.L, axis=1) # [N, F, 2]
+        self.set_model("checkpoints/LayerAnimate-Mix")
 
     @spaces.GPU
     def set_seed(self, seed):
@@ -243,6 +244,9 @@ class LayerAnimate:
         return output_video_path, output_video_traj_path
 
 
+layeranimate = LayerAnimate()
+
+
 def update_layer_region(image, layer_mask):
     if image is None or layer_mask is None:
         return None, False
@@ -412,7 +416,6 @@ if __name__ == "__main__":
                     6. **Note: Remember to click "Clear" button to clear all the controls before switching to another example.**<br>
                     """)
 
-        layeranimate = LayerAnimate()
         layer_indices = [gr.Number(value=i, visible=False) for i in range(LAYER_CAPACITY)]
         layer_tracking_points = [gr.State([[]]) for _ in range(LAYER_CAPACITY)]
         layer_masks = []
@@ -505,10 +508,9 @@ if __name__ == "__main__":
                 pretrained_model_path = gr.Dropdown(
                     label="Pretrained Model",
                     choices=[
-                        "None",
                         "checkpoints/LayerAnimate-Mix",
                     ],
-                    value="None",
+                    value="checkpoints/LayerAnimate-Mix",
                 )
                 text_prompt = gr.Textbox(label="Text Prompt", value="an anime scene.")
                 text_n_prompt = gr.Textbox(label="Negative Text Prompt", value="")