Hugging Face's logo

Spaces:
caltech-animal-tracking
/
Primate-Detection-GPU
Sleeping

App Files Community
Primate-Detection-GPU / README.md
annading's picture
annading
updated readme for local install instructions
f581713
preview code
|
raw history blame contribute delete
No virus
1.12 kB

A newer version of the Gradio SDK is available: 4.36.1

Upgrade
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
  1. Install the requirements using conda and pip.
conda create -n primates
conda activate primates
pip install -r requirements.txt

  1. 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.

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

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

python3 app.py