cjli commited on
Commit
5809b0d
·
1 Parent(s): e8e2ec7

fix example script

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -54,9 +54,12 @@ s2t = Speech2Text.from_pretrained(
54
  task_sym=task, # <pr>, <asr>, <g2p>, <p2g>
55
  )
56
 
57
- speech, rate = sf.read("sample.wav", sr=16000)
58
  prompt = "<na>" # G2P: set to ASR transcript; P2G: set to phone transcription with slashes
59
  pred = s2t(speech, text_prev=prompt)[0][0]
 
 
 
60
  if task == '<pr>' or task == '<g2p>:
61
  pred = pred.replace("/", "")
62
  print(pred)
 
54
  task_sym=task, # <pr>, <asr>, <g2p>, <p2g>
55
  )
56
 
57
+ speech, rate = sf.read("sample.wav")
58
  prompt = "<na>" # G2P: set to ASR transcript; P2G: set to phone transcription with slashes
59
  pred = s2t(speech, text_prev=prompt)[0][0]
60
+
61
+ # post-processing for better format
62
+ pred = pred.split("<notimestamps>")[1].strip()
63
  if task == '<pr>' or task == '<g2p>:
64
  pred = pred.replace("/", "")
65
  print(pred)