Slava917 commited on
Commit
311f387
1 Parent(s): df48e6f

change to return list

Browse files
Files changed (1) hide show
  1. predict.py +2 -1
predict.py CHANGED
@@ -21,7 +21,8 @@ def convert_probs(probs):
21
  if ids[i-1] != ids[i]:
22
  new = vocab[ids[i]]
23
  if new: s.append(new)
24
- return '.'.join(s)
 
25
 
26
  def predict(path):
27
  audio = loader(path)
21
  if ids[i-1] != ids[i]:
22
  new = vocab[ids[i]]
23
  if new: s.append(new)
24
+ #return '.'.join(s)
25
+ return s
26
 
27
  def predict(path):
28
  audio = loader(path)