Update app.py

app.py

@@ -437,9 +437,9 @@ def get_sam_mask(prompt_state, frames, frame_idx, input_points, input_labels):
     :return: (H, W) boolean mask, (H, W) float32 logits (or None), prompt_state
     """
 
-    model_cfg = "sam2_hiera_s.yaml"
+    model_cfg = "sam2_hiera_l.yaml"
     ckpt_path = "./Wan2.2-Animate-14B/process_checkpoint"
-    sam2_checkpoint_path = os.path.join(ckpt_path, 'sam2/sam2_hiera_small.pt')
+    sam2_checkpoint_path = os.path.join(ckpt_path, 'sam2/sam2_hiera_large.pt')
     
     video_predictor_local = build_sam2_video_predictor(model_cfg, sam2_checkpoint_path, device="cpu")    
     inference_state = video_predictor_local.init_state(images=np.array(frames), device="cpu")
@@ -526,6 +526,9 @@ def animate_scene(input_video, max_duration_s, edited_frame, rc_str,
         if "zerogpu quotas" not in err and "pro gpu quota" not in err:
             raise
 
+        if max_duration_s > 2:
+            raise
+
         max_duration_s = -1
 
         try:
@@ -797,7 +800,7 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 """)
                 input_video = gr.Video(label="Input Video", height=512)
 
-                max_duration_slider = gr.Slider(2, 4, 2, step=2, label="Max Duration", visible=False)
+                max_duration_slider = gr.Slider(2, 6, 2, step=2, label="Max Duration", visible=True)
 
                 gr.Examples(
                     examples=[