Update app.py

app.py

@@ -51,7 +51,7 @@ def predict(model, image_tensor):
         pred = torch.argmax(probs).item()
     return probs, pred
 
-def unlearn(model, image_tensor, label_idx, learning_rate, steps=20):
+def unlearn(model, image_tensor, label_idx, learning_rate, steps=10):
     """
     Performs targeted unlearning by updating only the final fully connected layer.
     The negative cross-entropy loss drives the confidence for the target class down.