README.md

metadata

title: Primate Detection Gradio
emoji: 🐢
colorFrom: red
colorTo: indigo
sdk: gradio
sdk_version: 4.23.0
app_file: app.py
pinned: false
license: apache-2.0

Primate Detection

Setup

1. Clone this repository through your local terminal by entering:

git lfs install 
git clone https://huggingface.co/spaces/caltech-animal-tracking/Primate-Detection-GPU

2. Install the requirements using conda and pip.

conda create -n primates
conda activate primates
pip install -r requirements.txt

3. Change the environment variables to fit your system.

   In app.py change BATCH_SIZE and CUDA_PATH to reflect current CUDA version and the size of your GPU. You can find your CUDA version by navigating to /usr/local on your computer. You may need to test a few different batch sizes to find one that works for your computer.

4. (Optional) Install FFmpeg if it's missing on your system by following the directions here.

5. Run app.py to install GroundingDINO locally and load the interface.

python3 app.py