Update app.py

app.py

@@ -35,7 +35,7 @@ def predict(model, image_tensor):
         pred = torch.argmax(probs).item()
     return probs, pred
  
-def unlearn(model, image_tensor, label_idx, learning_rate, steps=10):
+def unlearn(model, image_tensor, label_idx, learning_rate, steps=20):
     model.train()
     for m in model.modules():
         if isinstance(m, nn.BatchNorm2d):
@@ -126,7 +126,7 @@ demo = gr.Interface(
     fn=run_unlearning,
     inputs=[
         gr.Slider(0, len(trainset)-1, step=1, label="Select Index to Unlearn"),
-        gr.Slider(0.0001, 0.05, step=0.0001, value=0.005, label="Learning Rate (for Unlearning)")
+        gr.Slider(0.0001, 0.01, step=0.0001, value=0.005, label="Learning Rate (for Unlearning)")
     ],
     outputs="text",
     title="🔁 CIFAR-10 Machine Unlearning",