File size: 563 Bytes
8332d88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
#!/bin/bash
# Setup script for Hugging Face Spaces
# Install system dependencies
apt-get update && apt-get install -y git build-essential
# Downgrade setuptools
pip install setuptools==69.5.1
# Install PyTorch
pip install torch==2.0.1+cpu torchvision==0.15.2+cpu --index-url https://download.pytorch.org/whl/cpu
# Install detectron2 from source
git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
git checkout v0.6
pip install --no-build-isolation .
cd ..
rm -rf detectron2
# Install other requirements
pip install -r requirements.txt
|