Update model/train.py

Added code to properly save pipeline after creation which was missing and thus the reason for showing "Pipeline Available : False" in the frontend

model/train.py

@@ -222,6 +222,11 @@ class RobustModelTrainer:
             ('feature_select', feature_selector),
             ('model', None)  # Will be set during training
         ])
+
+        # Save the pipeline to .pkl format
+        joblib.dump(pipeline, "/tmp/pipeline.pkl")  # Save complete pipeline
+        joblib.dump(pipeline.named_steps['model'], "/tmp/model.pkl")  # Individual model
+        joblib.dump(pipeline.named_steps['vectorize'], "/tmp/vectorizer.pkl")  # Individual vectorizer
         
         return pipeline