fix: update inference main, tflite docs

TensorSpeech · May 26, 2024 · c8d9f06 · c8d9f06
1 parent 63a400c
commit c8d9f06
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/docs/tutorials/tflite.md b/docs/tutorials/tflite.md
@@ -11,7 +11,7 @@
 ## Conversion
 
 ```bash
-python3 examples/train.py \
+python3 examples/tflite.py \
  --config-path=/path/to/config.yml.j2 \
  --h5=/path/to/weight.h5 \
  --bs=1 \ # Batch size

diff --git a/examples/inferences/main.py b/examples/inferences/main.py
@@ -55,11 +55,8 @@ def main(
  previous_decoder_states=model.get_initial_decoder_states(),
  )
  )
- print(outputs.tokens)
  transcript = tokenizer.detokenize(outputs.tokens)[0].numpy().decode("utf-8")
-
  logger.info(f"Transcript: {transcript}")
- return transcript
 
 
 if __name__ == "__main__":