Update app.py

app.py

@@ -50,7 +50,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
     else:
         joint_attention_kwargs = None
 
-    apply_cache_on_pipe(
+    cache_scope = apply_cache_on_pipe(
         pipe,
         # residual_diff_threshold=0.2,
     )
@@ -67,6 +67,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             output_type="pil",
             good_vae=good_vae,  # Assuming good_vae is defined elsewhere
             joint_attention_kwargs=joint_attention_kwargs,  # Fixed parameter name
+            cache_scope=cache_scope,
         ):
             yield img, seed
     finally: