#!/bin/bash
set -e

# Install system dependencies for OpenCV
apt-get update && apt-get install -y libgl1 libglib2.0-0 wget curl

# Install Python packages
pip install datasets==2.21.0 ultralytics huggingface_hub pyyaml Pillow

# Download and run training script
wget -O /app/train_ppe_improved_v2.py https://huggingface.co/baskarmother/ppe-training-scripts/resolve/main/train_ppe_improved_v2.py

# Run training
cd /app
python train_ppe_improved_v2.py