Update train.py

train.py

@@ -1,6 +1,11 @@
 import torch
 from transformers import T5ForConditionalGeneration, T5Tokenizer, Trainer, TrainingArguments
 from datasets import load_dataset
+import os
+
+os.environ["HF_HOME"] = "/app/hf_cache"
+os.environ["HF_DATASETS_CACHE"] = "/app/hf_cache"
+os.environ["TRANSFORMERS_CACHE"] = "/app/hf_cache"
 
 # Load dataset (Replace this with your dataset)
 dataset = load_dataset("tatsu-lab/alpaca")  # Example alternative dataset