Update app.py

app.py

@@ -107,7 +107,7 @@ def generate_video(input_image, prompt, height, width,
                    pattern="dense", dense_layers=1, dense_timesteps=12, decay_factor=0.2,
                    progress=gr.Progress(track_tqdm=True)):
 
-    # from radial_attn.models.wan.inference import replace_wan_attention
+    from radial_attn.models.wan.inference import replace_wan_attention
     
     if input_image is None:
         raise gr.Error("Please upload an input image.")
@@ -122,17 +122,17 @@ def generate_video(input_image, prompt, height, width,
     # --- APPLY RADIAL ATTENTION PATCH ---
     # This function is called on every generation to ensure the correct settings
     # for either 'dense' or 'radial' attention are applied.
-    # with CalculateTime("Replace attention"):
-    #    replace_wan_attention(
-    #        pipe=pipe,
-    #        height=target_h,
-    #        width=target_w,
-    #        num_frames=num_frames,
-    #        dense_layers=dense_layers,
-    #        dense_timesteps=dense_timesteps,
-    #        decay_factor=decay_factor,
-    #        sparsity_type=pattern,
-    #    )
+    with CalculateTime("Replace attention"):
+        replace_wan_attention(
+            pipe=pipe,
+            height=target_h,
+            width=target_w,
+            num_frames=num_frames,
+            dense_layers=dense_layers,
+            dense_timesteps=dense_timesteps,
+            decay_factor=decay_factor,
+            sparsity_type=pattern,
+        )
 
     resized_image = input_image.resize((target_w, target_h))