Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ This repository provides an optimized JAX model for the Indic Whisper Model, bui
|
|
44 |
from whisper_jax import FlaxWhisperForConditionalGeneration, FlaxWhisperPipline
|
45 |
import jax.numpy as jnp
|
46 |
|
47 |
-
pipeline = FlaxWhisperPipline('parthiv11/
|
48 |
transcript= pipeline('sample.mp3')
|
49 |
|
50 |
```
|
|
|
44 |
from whisper_jax import FlaxWhisperForConditionalGeneration, FlaxWhisperPipline
|
45 |
import jax.numpy as jnp
|
46 |
|
47 |
+
pipeline = FlaxWhisperPipline('parthiv11/indic_whisper_nodcil', dtype=jnp.bfloat16)
|
48 |
transcript= pipeline('sample.mp3')
|
49 |
|
50 |
```
|