File size: 451 Bytes
78785c8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | #!/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
|