jaekookang commited on
Commit
d722032
1 Parent(s): e80f2b5

fixed type issue

Browse files
gradio_asr_en_libri100_word_vs_bpe.py CHANGED
@@ -46,10 +46,12 @@ def predict(wav_file):
46
  # Run inference
47
  W = model_word(speech)[0]
48
  B = model_bpe(speech)[0]
49
- comparison = list(d.compare([W+'\n'], [B+'\n']))
 
 
50
 
51
  logger.info('predicted')
52
- return W[0], B[0], comparison
53
 
54
  iface = gr.Interface(
55
  predict,
 
46
  # Run inference
47
  W = model_word(speech)[0]
48
  B = model_bpe(speech)[0]
49
+ word_decoded = W[0]
50
+ bpe_decoded = B[0]
51
+ comparison = list(d.compare([word_decoded+'\n'], [bpe_decoded+'\n']))
52
 
53
  logger.info('predicted')
54
+ return word_decoded, bpe_decoded, comparison
55
 
56
  iface = gr.Interface(
57
  predict,