Update app.py

app.py

@@ -118,6 +118,7 @@ def run_unlearning(index_to_unlearn, learning_rate):
 
     # Prediction after unlearning
     probs_after, pred_after = predict(unlearned_model, image_tensor)
+    print(probs_after)
     conf_after = probs_after[label_idx].item()
 
     # Evaluate full test set performance on both models