Update app.py

app.py

@@ -120,6 +120,8 @@ def simple_generative_inference(image, mode, model, n_iterations=10, step_size=0
         # Update image tensor
         image_tensor = image_tensor + step_size * grad
         image_tensor = torch.clamp(image_tensor, -eps, eps)   # Keep within range
+        image_tensor = image_tensor.clone().detach().requires_grad_(True)  # Ensure it's a new leaf tensor
+        image_tensor.retain_grad()
 
     # Generate gradient visualization
     grad_image = grad.abs().mean(dim=1).squeeze().cpu().numpy()