model/train_model.py

import os
# Set the Hugging Face cache directory to a writable location
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
from model import train_sms_classifier

# Train the model and save it
train_sms_classifier()