Update README.md
Browse files
README.md
CHANGED
|
@@ -97,8 +97,8 @@ Currently, Dolphin-Fangyan provides multiple model sizes optimized for different
|
|
| 97 |
|:------:|:----------:|:----------:|
|
| 98 |
| base.fangyan | 74 M | ❌ |
|
| 99 |
| base.fangyan.streaming | 74 M |❌ |
|
| 100 |
-
| small.fangyan | 0.4 B |
|
| 101 |
-
| small.fangyan.streaming | 0.4 B |
|
| 102 |
| small.fangyan.prompt | 0.4 B | Prompt-based Hotwords |
|
| 103 |
|
| 104 |
|
|
@@ -192,8 +192,7 @@ import dolphin
|
|
| 192 |
from dolphin import transcribe
|
| 193 |
|
| 194 |
model_name = 'small.fangyan'
|
| 195 |
-
model = dolphin.load_model(model_name, f"/
|
| 196 |
-
# model = dolphin.load_model(model_name, f"/data/models/dolphin/{model_name}", "cpu")
|
| 197 |
|
| 198 |
result = transcribe(model, 'audio.wav')
|
| 199 |
print(result.text)
|
|
@@ -209,7 +208,7 @@ print(result.text)
|
|
| 209 |
## prompt-based hotwords
|
| 210 |
|
| 211 |
model_name = 'small.fangyan.prompt'
|
| 212 |
-
model = dolphin.load_model(model_name, f"/
|
| 213 |
|
| 214 |
result = transcribe(model, 'audio.wav', hotwords=['诺香丹青牌科研胶囊'], use_prompt_hotword=True, use_two_stage_filter=True, decoding_method='attention')
|
| 215 |
|
|
|
|
| 97 |
|:------:|:----------:|:----------:|
|
| 98 |
| base.fangyan | 74 M | ❌ |
|
| 99 |
| base.fangyan.streaming | 74 M |❌ |
|
| 100 |
+
| small.fangyan | 0.4 B | Encoder-biased Hotwords |
|
| 101 |
+
| small.fangyan.streaming | 0.4 B | Encoder-biased Hotwords |
|
| 102 |
| small.fangyan.prompt | 0.4 B | Prompt-based Hotwords |
|
| 103 |
|
| 104 |
|
|
|
|
| 192 |
from dolphin import transcribe
|
| 193 |
|
| 194 |
model_name = 'small.fangyan'
|
| 195 |
+
model = dolphin.load_model(model_name, f"/data/models/dolphin/{model_name}", "cuda")
|
|
|
|
| 196 |
|
| 197 |
result = transcribe(model, 'audio.wav')
|
| 198 |
print(result.text)
|
|
|
|
| 208 |
## prompt-based hotwords
|
| 209 |
|
| 210 |
model_name = 'small.fangyan.prompt'
|
| 211 |
+
model = dolphin.load_model(model_name, f"/data/models/dolphin/{model_name}", "cuda")
|
| 212 |
|
| 213 |
result = transcribe(model, 'audio.wav', hotwords=['诺香丹青牌科研胶囊'], use_prompt_hotword=True, use_two_stage_filter=True, decoding_method='attention')
|
| 214 |
|