DataoceanAI commited on
Commit
6a62653
·
verified ·
1 Parent(s): f217fdd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -5
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 | Encode-biased Hotwords |
101
- | small.fangyan.streaming | 0.4 B | Encode-biased Hotwords |
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"/home/duhu/.cache/dolphin/{model_name}", "cpu")
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"/home/duhu/.cache/dolphin/{model_name}", "cpu")
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