#!/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