From 2ae6193b0315d0de329ea22e4e7f8d843bdf9909 Mon Sep 17 00:00:00 2001
From: Aly Sewelam <linlysmolworthy@gmail.com>
Date: Thu, 30 Jul 2026 17:48:30 +0300
Subject: [PATCH] fix: properly save model

---
 lab4/train_spam_model.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/lab4/train_spam_model.py b/lab4/train_spam_model.py
index c4d849b..af23a4d 100644
--- a/lab4/train_spam_model.py
+++ b/lab4/train_spam_model.py
@@ -65,11 +65,10 @@ def main():
     for text, label, score in zip(samples, preds, scores_demo):
         verdict = "SPAM" if label == 1 else "HAM"
         print(f"[{verdict}] score={score:.2f} :: {text}")
-
-
-if __name__ == "__main__":
     # persist our model
     joblib.dump(vectorizer, "vectorizer.joblib")
     joblib.dump(clf, "classifier.joblib")
+
+if __name__ == "__main__":
     print("\nSaved model artifacts.")
     main()