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__":