Update app.py

app.py

@@ -71,8 +71,11 @@ def simple_generative_inference(image, mode, model, n_iterations=10, step_size=0
         transforms.ToTensor(),
         transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
     ])
-    image_tensor = transform(image).unsqueeze(0).requires_grad_(True)
-    image_tensor.retain_grad()  # Ensure gradients are retained for non-leaf tensor
+    image_tensor = transform(image).unsqueeze(0)
+
+    image_tensor = image_tensor.detach().clone()
+    image_tensor.requires_grad = True
+    # image_tensor.retain_grad()  # Ensure gradients are retained for non-leaf tensor
 
     if mode == "increase confidence":
         output = model(image_tensor)
@@ -85,7 +88,6 @@ def simple_generative_inference(image, mode, model, n_iterations=10, step_size=0
         
     for _ in range(n_iterations):
 
-        image_tensor
         # Zero gradients
         model.zero_grad()