Skip to content

Commit

Permalink
📰 Adds readme instructions
Browse files Browse the repository at this point in the history
  • Loading branch information
mrharpo committed May 8, 2024
1 parent 7278c4d commit 067d22d
Showing 1 changed file with 46 additions and 0 deletions.
46 changes: 46 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,3 +1,49 @@
![deploy](https://github.com/WGBH-MLA/whisper-bot/actions/workflows/CD.yml/badge.svg)

# whisper-bot

This repository provides a pre-built docker image for OpenAI's Whisper model.

## Usage

To use the model, you can pull the pre-built docker image, or build the image yourself.

### Pre-built image

```bash
docker pull ghcr.io/wgbh-mla/whisper-bot:latest
```

### Build the image

```bash
docker build -t whisper-bot .
```

## Running Whisper

```bash
docker run --rm -itv $(pwd):/root -v $HOME/.cache/whisper/:/root/.cache/whisper/ ghcr.io/wgbh-mla/whisper-bot:latest whisper [WHISPER_ARGS] FILENAME
```

### Whisper Arguments

- `--model`: The model to use. Defaults to `base`.
- Options: `tiny`, `base`, `small`, `medium`, `large`
- `--language`: The language to use. Defaults to `en`.

The full list of arguments can be found by running `whisper --help`

## ARM Support

If you are running on an ARM machine (including the Mac M series processors), use the `arm64-main` tag to get an ARM optimized image.

## Distributed whisper-bot

See the [whisper-bot](whisper-bot) directory for an example of running whisper-bot in a distributed system.

**Note**: _This script is specific to the GBH MLA environment and should only be used an an example._

## Credits

Created by [GBH Media Library & Archives](https://www.wgbh.org/foundation/archives)

0 comments on commit 067d22d

Please sign in to comment.