Create train.sh

train.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+
+# Log in to Hugging Face (you'll need to enter a token the first time)
+huggingface-cli login --token $HF_TOKEN
+
+# Run AutoTrain on Llama 3.1
+autotrain llm \
+  --train \
+  --model meta-llama/Meta-Llama-3-8B \
+  --data_path /app/news_data.csv \
+  --output_dir /app/model_output \
+  --epochs 3 \
+  --batch_size 2
+
+# Push the trained model to Hugging Face Hub
+autotrain llm \
+  --push_to_hub \
+  --model_dir /app/model_output \
+  --repo_name your_username/llama3-news-recommendation