From c8d9f06fa70011a560453c4027bb8b97e9b8d780 Mon Sep 17 00:00:00 2001
From: Huy Le Nguyen <nlhuy.cs.16@gmail.com>
Date: Sun, 26 May 2024 21:19:13 +0700
Subject: [PATCH] fix: update inference main, tflite docs

---
 docs/tutorials/tflite.md    | 2 +-
 examples/inferences/main.py | 3 ---
 2 files changed, 1 insertion(+), 4 deletions(-)

diff --git a/docs/tutorials/tflite.md b/docs/tutorials/tflite.md
index 25f29b0d9..6bb02bd4e 100644
--- 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
index 393ab20fd..d381c0d99 100644
--- 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__":