etchen commited on
Commit
3686641
·
verified ·
1 Parent(s): 5da095f

Update TaiwaneseHokkien.py

Browse files
Files changed (1) hide show
  1. TaiwaneseHokkien.py +2 -2
TaiwaneseHokkien.py CHANGED
@@ -4,9 +4,9 @@ from transformers import pipeline
4
  model_repo_id = "emlinking/wav2vec2-large-xls-r-300m-tsm-asr-v6"
5
  util = Converter()
6
  def get_initial_length(s):
7
- if s[2] == 'h':
8
  return 3
9
- elif s[1] in ['s', 'h']:
10
  return 2
11
  else:
12
  return 1
 
4
  model_repo_id = "emlinking/wav2vec2-large-xls-r-300m-tsm-asr-v6"
5
  util = Converter()
6
  def get_initial_length(s):
7
+ if len(s) >= 3 and s[2] == 'h':
8
  return 3
9
+ elif len(s) >= 2 and s[1] in ['s', 'h']:
10
  return 2
11
  else:
12
  return 1