File size: 3,276 Bytes
64c9f84
 
 
8e676e0
fc4d32d
64c9f84
 
fc4d32d
64c9f84
8e676e0
 
 
64c9f84
8e676e0
 
64c9f84
 
dbc614f
8e676e0
 
 
dbc614f
8e676e0
fc4d32d
8e676e0
 
 
 
 
 
 
fc4d32d
64c9f84
c667742
8e676e0
 
64c9f84
 
8e676e0
 
 
 
fc4d32d
8e676e0
c667742
8e676e0
 
fc4d32d
 
8e676e0
 
 
 
 
64c9f84
8e676e0
fc4d32d
8e676e0
 
 
fc4d32d
 
8e676e0
 
 
fc4d32d
8e676e0
 
fc4d32d
c667742
8e676e0
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
---
pipeline_tag: automatic-speech-recognition
license: mit
base_model: UsefulSensors/moonshine-tiny
library_name: zeromodels
tags:
- keras
- zeromodels
- moonshine
- automatic-speech-recognition
- audio
- arxiv:2410.15608
- pytorch
- jax
- tf
---

## ***See [our collection](https://huggingface.co/collections/zeromodels/moonshine-6a8eaf317c682a7009c68b2c) for all versions of Moonshine.***

# Run Moonshine with Keras 3: JAX, PyTorch, or TensorFlow

[![GitHub](https://img.shields.io/badge/GitHub-ZeroModels-black?logo=github)](https://github.com/IMvision12/ZeroModels) [![Docs](https://img.shields.io/badge/Docs-Moonshine-blue)](https://imvision12.github.io/ZeroModels/moonshine/) [![Collection](https://img.shields.io/badge/HF-Moonshine%20collection-yellow)](https://huggingface.co/collections/zeromodels/moonshine-6a8eaf317c682a7009c68b2c)

# zeromodels/moonshine_tiny

Paper: [Moonshine: Speech Recognition for Live Transcription and Voice Commands (arXiv:2410.15608)](https://arxiv.org/abs/2410.15608) · [HF Papers](https://huggingface.co/papers/2410.15608)

Moonshine is an English ASR encoder-decoder built for **short / live** audio: the encoder sees the raw waveform length you pass in (no Whisper-style 30 s pad), so short commands stay cheap. Output is cased and punctuated.

For more details on the model, please go to the upstream [model card](https://huggingface.co/UsefulSensors/moonshine-tiny).

Pure-**Keras 3** conversion of [`UsefulSensors/moonshine-tiny`](https://huggingface.co/UsefulSensors/moonshine-tiny) for [zeromodels](https://github.com/IMvision12/ZeroModels). One implementation runs unmodified on **TensorFlow / Torch / JAX**.

This is an **ASR** checkpoint (`MoonshineConditionalGenerate`).

## ✨ Quick start

```python
import os
os.environ["KERAS_BACKEND"] = "torch"  # or "jax" / "tensorflow"

import soundfile as sf
from zeromodels.models.moonshine import (
    MoonshineProcessor,
    MoonshineConditionalGenerate,
)

model = MoonshineConditionalGenerate.from_weights("zeromodels/moonshine_tiny")
processor = MoonshineProcessor.from_weights("zeromodels/moonshine_tiny")

audio, sr = sf.read("your_audio.wav", dtype="float32")  # 16 kHz mono
# Cost scales with clip length: no fixed 30 s pad like Whisper.
text = model.generate(audio, processor)
print(repr(text[0]))
```

Load any Moonshine variant the same way with `from_weights("zeromodels/<variant>")`:

| Variant | Hub |
|---|---|
| `moonshine_tiny` | [`zeromodels/moonshine_tiny`](https://huggingface.co/zeromodels/moonshine_tiny) |
| `moonshine_base` | [`zeromodels/moonshine_base`](https://huggingface.co/zeromodels/moonshine_base) |

## Tips

- Set `KERAS_BACKEND` **before** importing Keras / zeromodels.
- Prefer `MoonshineProcessor.from_weights(...)` so feature extraction matches.
- English-only; pass a list of waveforms to batch.
- See [Moonshine docs](https://imvision12.github.io/ZeroModels/moonshine/) and [Loading Weights](https://imvision12.github.io/ZeroModels/loading_weights/).
- Community / upstream safetensors still work via the `hf:` prefix, e.g. `MoonshineConditionalGenerate.from_weights("hf:UsefulSensors/moonshine-tiny")`.

## Special Thanks

A huge thank you to the Useful Sensors Moonshine authors for creating and releasing these models.

License: MIT.