Update app.py

app.py

@@ -143,7 +143,7 @@ def run_unlearning(index_to_unlearn, learning_rate):
 - Original Model: {orig_acc:.2f}% accuracy, Loss: {orig_loss:.4f}
 - Unlearned Model: {unlearn_acc:.2f}% accuracy, Loss: {unlearn_loss:.4f}
 """
-    return result, probs_after
+    return result
 
 # Gradio Interface for interactive unlearning demonstration
 demo = gr.Interface(