Update app.py

app.py

@@ -204,7 +204,7 @@ def initialize_model():
                     _reset_cache_fresh()
 
         # Sample with the fresh noise (matching notebook: sample(model, noise, actions[:, aidx:aidx+1], ...))
-        z = sample(model_, noise, action_buf, num_steps=n_steps, cfg=cfg, negative_actions=None)
+        z = sample(model_, noise, action_buf, num_steps=n_steps, cfg=cfg, negative_actions=None, cache=cache)
         
         if clamp:
             z = t.clamp(z, -1, 1)